structure

nile4000 · Aug 3, 2024 · 2caf535 · 2caf535
1 parent 4d254c8
commit 2caf535
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/app.py b/app.py
@@ -24,13 +24,16 @@
 app = FastAPI()
 
 # Configure CORS
-origins = ["*"]
+ORIGINS = ["*"] #Cors-disabled
+METHODS = ["POST"]
+HEADERS = ["*"]
+
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=origins,
+    allow_origins=ORIGINS,
     allow_credentials=True,
-    allow_methods=["POST"],
-    allow_headers=["*"],
+    allow_methods=METHODS,
+    allow_headers=HEADERS,
 )
 
 # Detect host device for torch
@@ -49,6 +52,7 @@
 
 # Create model pipeline
 MODEL_ID = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+
 pipe = pipeline(
     "text-generation",
     model=MODEL_ID,