Update main.py

main.py

@@ -4,14 +4,17 @@ from firebase_admin import credentials, firestore
 from transformers import pipeline
 from pydantic import BaseModel
 from typing import List, Optional
+from huggingface_hub import login
+from fastapi.responses import JSONResponse
 
 # Load Firebase
 cred = credentials.Certificate("firebase_config.json")
 firebase_admin.initialize_app(cred)
 db = firestore.client()
 
-login(os.getenv("HF_TOKEN"))
 # Load AI Model
+login(os.getenv("HF_TOKEN"))
+
 ai_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
 
 app = FastAPI()