added_main

app.py

@@ -19,7 +19,6 @@ from fastapi.middleware.cors import CORSMiddleware
 app = FastAPI()
 
 app.add_middleware(
-
     CORSMiddleware,
     allow_origins=["*"],
     allow_credentials=True,
@@ -71,6 +70,7 @@ try:
     # Move model to device
     combined_model = combined_model.to(device)
     combined_model.eval()
+    print("Models loaded successfully!")
 except Exception as e:
     raise SystemExit(f"Error loading models: {e}")
 
@@ -138,4 +138,7 @@ async def predict(files: list[UploadFile]):
         return JSONResponse(
             status_code=500,
             content={"error": f"Error processing request: {str(e)}"}
-        )
+        )
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)