chengwang96 commited on
Commit
3f31363
·
1 Parent(s): 0ad9500

start with sample

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -8,15 +8,18 @@ ENHANCE_INPUT_DIR = "sr_input"
8
  ENHANCE_OUTPUT_DIR = "sr_output"
9
  ENHANCE_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 30)]
10
  ENHANCE_CHOICES = ["请选择样例图片或者上传"] + ENHANCE_IMG_NAMES
 
11
 
12
  SEG_INPUT_DIR = "seg_input"
13
  SEG_OUTPUT_DIR = "seg_output"
14
  SEG_IMG_NAMES = [f"img_{i}.png" for i in range(1, 30)]
15
  SEG_CHOICES = ["请选择样例图片或者上传"] + SEG_IMG_NAMES
 
16
 
17
  DIAG_INPUT_DIR = "diag_input"
18
  DIAG_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 32)]
19
  DIAG_CHOICES = ["请选择样例图片或者上传"] + DIAG_IMG_NAMES
 
20
 
21
  KVASIR_CLASSES = [
22
  "正常粘膜 (normal-z-line)",
@@ -252,7 +255,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
252
  gr.HTML("<div class='big-title'>图像增强模块</div>")
253
  enhance_select = gr.Dropdown(
254
  choices=ENHANCE_CHOICES,
255
- value="---------------",
256
  label="请选择样例图片或者上传",
257
  filterable=True
258
  )
@@ -299,7 +302,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
299
  gr.HTML("<div class='big-title'>图像分割模块</div>")
300
  seg_select = gr.Dropdown(
301
  choices=SEG_CHOICES,
302
- value="---------------",
303
  label="请选择样例图片或者上传",
304
  filterable=True
305
  )
@@ -337,7 +340,7 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
337
  with gr.Column():
338
  diag_select = gr.Dropdown(
339
  choices=DIAG_CHOICES,
340
- value="---------------",
341
  label="请选择样例图片或者上传",
342
  filterable=True
343
  )
@@ -374,4 +377,12 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
374
  outputs=[diag_select, diag_input_img, diag_table, diag_result, diag_suggestion]
375
  )
376
 
 
 
 
 
 
 
 
 
377
  demo.launch()
 
8
  ENHANCE_OUTPUT_DIR = "sr_output"
9
  ENHANCE_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 30)]
10
  ENHANCE_CHOICES = ["请选择样例图片或者上传"] + ENHANCE_IMG_NAMES
11
+ ENHANCE_DEFAULT = ENHANCE_IMG_NAMES[0]
12
 
13
  SEG_INPUT_DIR = "seg_input"
14
  SEG_OUTPUT_DIR = "seg_output"
15
  SEG_IMG_NAMES = [f"img_{i}.png" for i in range(1, 30)]
16
  SEG_CHOICES = ["请选择样例图片或者上传"] + SEG_IMG_NAMES
17
+ SEG_DEFAULT = SEG_IMG_NAMES[0]
18
 
19
  DIAG_INPUT_DIR = "diag_input"
20
  DIAG_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 32)]
21
  DIAG_CHOICES = ["请选择样例图片或者上传"] + DIAG_IMG_NAMES
22
+ DIAG_DEFAULT = DIAG_IMG_NAMES[0]
23
 
24
  KVASIR_CLASSES = [
25
  "正常粘膜 (normal-z-line)",
 
255
  gr.HTML("<div class='big-title'>图像增强模块</div>")
256
  enhance_select = gr.Dropdown(
257
  choices=ENHANCE_CHOICES,
258
+ value=ENHANCE_DEFAULT,
259
  label="请选择样例图片或者上传",
260
  filterable=True
261
  )
 
302
  gr.HTML("<div class='big-title'>图像分割模块</div>")
303
  seg_select = gr.Dropdown(
304
  choices=SEG_CHOICES,
305
+ value=SEG_DEFAULT,
306
  label="请选择样例图片或者上传",
307
  filterable=True
308
  )
 
340
  with gr.Column():
341
  diag_select = gr.Dropdown(
342
  choices=DIAG_CHOICES,
343
+ value=DIAG_DEFAULT,
344
  label="请选择样例图片或者上传",
345
  filterable=True
346
  )
 
377
  outputs=[diag_select, diag_input_img, diag_table, diag_result, diag_suggestion]
378
  )
379
 
380
+ # ====== 页面加载时自动显示默认结果 ======
381
+ demo.load(fn=show_enhance_input, inputs=gr.State(ENHANCE_DEFAULT), outputs=enhance_input_img)
382
+ demo.load(fn=enhance_demo, inputs=gr.State(ENHANCE_DEFAULT), outputs=enhance_output_img)
383
+ demo.load(fn=show_seg_input, inputs=gr.State(SEG_DEFAULT), outputs=seg_input_img)
384
+ demo.load(fn=segment_demo, inputs=gr.State(SEG_DEFAULT), outputs=seg_output_img)
385
+ demo.load(fn=show_diag_input, inputs=gr.State(DIAG_DEFAULT), outputs=diag_input_img)
386
+ demo.load(fn=diagnose_demo, inputs=gr.State(DIAG_DEFAULT), outputs=[diag_table, diag_result, diag_suggestion])
387
+
388
  demo.launch()