Update app.py

app.py

@@ -1,29 +1,19 @@
 import gradio as gr
 from hf_diffusion_service import HFDiffusionService
 
-# Initialize the service
-try:
-    service = HFDiffusionService()
-except Exception as e:
-    print(f"Failed to initialize HFDiffusionService: {e}")
-    service = None
+service = HFDiffusionService()
 
-# Define Gradio function
 def generate_ct(mask_image):
-    if service is None:
-        return "Error: Model failed to initialize."
-    result = service.generate_image(mask_image)
-    if result is None:
-        return "Error generating image."
-    return result
+    # Return the PIL image directly
+    return service.generate_image(mask_image)
 
-# Create Gradio interface
+# ✅ Single-interface Gradio app
 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="Upload or draw a segmentation mask to generate a synthetic CT scan using the conditional diffusion model."
+    description="Upload or draw a mask to generate a synthetic CT scan"
 )
 
 if __name__ == "__main__":