Update app.py

app.py

@@ -1,9 +1,3 @@
-import os
-os.environ["GRADIO_ENABLE_QUEUE"] = "1"  # ensures queue
-os.environ["GRADIO_SERVER_PORT"] = "7860"
-os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
-os.environ["GRADIO_SSR_MODE"] = "false"  # ✅ explicitly disable SSR
-
 import gradio as gr
 from hf_diffusion_service import HFDiffusionService
 from PIL import Image
@@ -11,15 +5,14 @@ from PIL import Image
 service = HFDiffusionService()
 
 def generate_ct(mask_image: Image.Image):
-    return service.generate_image(mask_image)
+    return service.generate_image_base64(mask_image)
 
 demo = gr.Interface(
     fn=generate_ct,
     inputs=gr.Image(type="pil", label="Segmentation Mask"),
     outputs=gr.Image(type="pil", label="Generated CT Scan"),
-    title="Conditional Diffusion Medical Image Generator",
-    description="Draw or upload a mask to generate a synthetic CT scan.",
-    flagging_mode="never"
+    title="Synthetic CT Generator",
+    description="Upload or draw a mask to generate a CT scan."
 )
 
 if __name__ == "__main__":