tan200224 commited on
Commit
b403fe1
·
verified ·
1 Parent(s): e8993b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -14
app.py CHANGED
@@ -1,32 +1,26 @@
 
 
 
 
 
 
1
  import gradio as gr
2
  from hf_diffusion_service import HFDiffusionService
3
  from PIL import Image
4
 
5
- # Initialize diffusion model service
6
  service = HFDiffusionService()
7
 
8
- # Function for CT scan generation
9
  def generate_ct(mask_image: Image.Image):
10
- """
11
- Takes a PIL mask image and returns a generated CT scan (PIL).
12
- """
13
  return service.generate_image(mask_image)
14
 
15
- # Gradio Interface
16
  demo = gr.Interface(
17
  fn=generate_ct,
18
  inputs=gr.Image(type="pil", label="Segmentation Mask"),
19
  outputs=gr.Image(type="pil", label="Generated CT Scan"),
20
  title="Conditional Diffusion Medical Image Generator",
21
- description="Draw or upload a mask to generate a synthetic CT scan using the conditional diffusion model.",
22
  flagging_mode="never"
23
  )
24
 
25
  if __name__ == "__main__":
26
- # ✅ Use only supported parameters
27
- demo.launch(
28
- server_name="0.0.0.0",
29
- server_port=7860,
30
- share=False,
31
- show_error=True
32
- )
 
1
+ import os
2
+ os.environ["GRADIO_ENABLE_QUEUE"] = "1" # ensures queue
3
+ os.environ["GRADIO_SERVER_PORT"] = "7860"
4
+ os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
5
+ os.environ["GRADIO_SSR_MODE"] = "false" # ✅ explicitly disable SSR
6
+
7
  import gradio as gr
8
  from hf_diffusion_service import HFDiffusionService
9
  from PIL import Image
10
 
 
11
  service = HFDiffusionService()
12
 
 
13
  def generate_ct(mask_image: Image.Image):
 
 
 
14
  return service.generate_image(mask_image)
15
 
 
16
  demo = gr.Interface(
17
  fn=generate_ct,
18
  inputs=gr.Image(type="pil", label="Segmentation Mask"),
19
  outputs=gr.Image(type="pil", label="Generated CT Scan"),
20
  title="Conditional Diffusion Medical Image Generator",
21
+ description="Draw or upload a mask to generate a synthetic CT scan.",
22
  flagging_mode="never"
23
  )
24
 
25
  if __name__ == "__main__":
26
+ demo.launch()