khaledadrani

from fastapi import FastAPI
from pydantic import BaseModel
from utils.nlp import extract_ents
from utils.twitter_api import get_response

app = FastAPI()

class Query(BaseModel):
    keyword: str
    max_results: int

khaledadrani

import spacy
import re

ner = spacy.load('trf_ner\model-best')

def test_model():
    '''
    Check if the model is loaded properly
    '''
     

khaledadrani

import os
from dotenv import load_dotenv
import requests 

#load your credentials through the .env file
load_dotenv()

def create_headers():    
    
    api_key = os.getenv('api_key')