Update app.py

app.py

@@ -3,8 +3,6 @@ import torch
 import torch.nn as nn
 from torchvision.utils import make_grid
 import gradio as gr
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse
 
 # ------------------------
 # CONFIG
@@ -90,7 +88,7 @@ with gr.Blocks(title="KYO") as demo:
         gr.Image(value=LOGO_PATH, show_label=False, width=180)
 
     gr.Markdown("# KYO")
-    gr.Markdown("Generate graffiti-style images using your trained PyTorch GAN.")
+    gr.Markdown("Generate graffiti-style images using PyTorch GAN.")
 
     with gr.Row():
         num_samples = gr.Slider(
@@ -115,28 +113,7 @@ with gr.Blocks(title="KYO") as demo:
     )
 
 # ------------------------
-# FastAPI app + cron endpoint
+# Launch app
 # ------------------------
-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"
-    })
-
-@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="/")
-
-# No uvicorn.run() needed; HF Spaces handles the server and port
+if __name__ == "__main__":
+    demo.launch()