Update app.py

app.py

@@ -5,7 +5,6 @@ from torchvision.utils import make_grid
 import gradio as gr
 from fastapi import FastAPI
 from fastapi.responses import JSONResponse
-import uvicorn
 
 # ------------------------
 # CONFIG
@@ -43,10 +42,8 @@ class Generator(nn.Module):
 def load_generator():
     if not os.path.exists(GENERATOR_PATH):
         raise FileNotFoundError(
-            f"Checkpoint not found: {GENERATOR_PATH}. "
-            "Make sure it exists in your Hugging Face Space."
+            f"Checkpoint not found: {GENERATOR_PATH}. Make sure it exists in your Hugging Face Space."
         )
-
     model = Generator().to(DEVICE)
     state_dict = torch.load(GENERATOR_PATH, map_location=DEVICE)
     model.load_state_dict(state_dict)
@@ -124,6 +121,7 @@ app = FastAPI()
 
 @app.get("/cron")
 def cron():
+    """Cron endpoint – just returns a message for now."""
     return JSONResponse({
         "ok": True,
         "message": "KYO cron endpoint is live"
@@ -131,14 +129,14 @@ def cron():
 
 @app.get("/health")
 def health():
+    """Health check endpoint."""
     return JSONResponse({
         "status": "healthy",
         "device": DEVICE,
         "model_loaded": G is not None
     })
 
+# Mount Gradio app on FastAPI
 app = gr.mount_gradio_app(app, demo, path="/")
 
-if __name__ == "__main__":
-    # Launch Gradio normally; HF Space handles ports
-    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+# No uvicorn.run() needed; HF Spaces handles the server and port