Update main.py

main.py

@@ -114,10 +114,10 @@ os.makedirs(OUTPUT_FOLDER, exist_ok=True)
 
 # Lightweight model configuration
 MODEL_NAME = "distilgpt2"
+MAX_FILE_SIZE = 2 * 1024 * 1024  # 2MB
 TIMEOUT = 10  # seconds
 MAX_ROWS = 100
 MAX_COLUMNS = 5
-MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
 try:
     visualization_model = pipeline(
         "text-generation",
@@ -739,4 +739,9 @@ if __name__ == "__main__":
         print(f"Error: {response.status_code}\n{response.text}")
 
     # 6. Optional: Run the server (if not already running)
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+ uvicorn.run(
+        app, 
+        host="0.0.0.0", 
+        port=8000,
+        timeout_keep_alive=15
+    )