Update main.py

main.py

@@ -3,6 +3,7 @@ from fastapi.responses import StreamingResponse
 from fastapi.responses import FileResponse, HTMLResponse
 import os
 import io
+from fastapi.middleware.cors import CORSMiddleware
 import google.generativeai as genai
 
 import torch
@@ -163,6 +164,15 @@ def refresh_model():
         print("model refreshed")
 
 app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 @app.post("/llm_response/")
 async def llm_response(chain,id,mode):
     id = int(id)