Update app.py

app.py

@@ -3,12 +3,12 @@ from HFDiffusionService import HFDiffusionService
 
 service = HFDiffusionService()
 
-def generate_from_mask(mask_image):
-    output_image = service.generate_image(mask_image)
-    return output_image
+def generate_ct(mask_image):
+    image = service.generate_image(mask_image)
+    return image
 
 demo = gr.Interface(
-    fn=generate_from_mask,
+    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"