Update app.py

app.py

@@ -25,7 +25,7 @@ app.add_middleware(
 )
 
 # Load your API key from the environment (defaults to "change_me")
-OLLAMA_API_KEY = os.environ.get("API_KEY", "change_me")
+API_KEY = os.environ.get("API_KEY", "change_me")
 
 @app.post("/api/generate")
 async def generate(request: Request):
@@ -38,7 +38,7 @@ async def generate(request: Request):
         raise HTTPException(status_code=401, detail="Missing or invalid Authorization header")
 
     token = auth_header.split(" ")[1]
-    if token != OLLAMA_API_KEY:
+    if token != API_KEY:
         raise HTTPException(status_code=401, detail="Invalid API key")
 
     # 2. Parse the request JSON
@@ -62,7 +62,7 @@ async def generate(request: Request):
             stderr=subprocess.PIPE,
             text=True
         )
-        output, error = process.communicate(timeout=60)  # Timeout after 60 seconds
+        output, error = process.communicate(timeout=600)  # Timeout after 60 seconds
 
         if process.returncode != 0:
             logger.error(f"Ollama Error: {error.strip()}")