Update app.py

app.py

@@ -1,16 +1,29 @@
 import gradio as gr
-from hf_diffusion_service import HFDiffusionService  # <-- updated
+from hf_diffusion_service import HFDiffusionService
 
-service = HFDiffusionService()
+# Initialize the service
+try:
+    service = HFDiffusionService()
+except Exception as e:
+    print(f"Failed to initialize HFDiffusionService: {e}")
+    service = None
 
+# Define Gradio function
 def generate_ct(mask_image):
-    return service.generate_image(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
 
+# Create Gradio interface
 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"
+    title="Conditional Diffusion Medical Image Generator",
+    description="Upload or draw a segmentation mask to generate a synthetic CT scan using the conditional diffusion model."
 )
 
 if __name__ == "__main__":