Update app.py

app.py

@@ -1,11 +1,10 @@
 import gradio as gr
-from HFDiffusionService import HFDiffusionService
+from hf_diffusion_service import HFDiffusionService  # <-- updated
 
 service = HFDiffusionService()
 
 def generate_ct(mask_image):
-    image = service.generate_image(mask_image)
-    return image
+    return service.generate_image(mask_image)
 
 demo = gr.Interface(
     fn=generate_ct,