jayn95 commited on
Commit
fdfb1fb
·
verified ·
1 Parent(s): d407986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -38,9 +38,9 @@ def detect(image, conf=0.4, iou=0.5):
38
  output = results[0].plot()[:, :, ::-1] # BGR → RGB
39
 
40
  if len(boxes) > 0:
41
- diagnosis = "⚠️ __DETECTION_TEXT__ detected."
42
  else:
43
- diagnosis = "🟢 No __DETECTION_TEXT__ detected."
44
 
45
  return [output, diagnosis]
46
 
@@ -50,14 +50,14 @@ interface = gr.Interface(
50
  fn=detect,
51
  inputs=[
52
  gr.Image(type="pil", label="Upload Image"),
53
- gr.Slider(0, 1, value=0.4, step=0.05, label="Confidence Threshold"),
54
  gr.Slider(0, 1, value=0.5, step=0.05, label="NMS IoU Threshold"),
55
  ],
56
  outputs=[
57
- gr.Image(label="Detection Result"),
58
  gr.Textbox(label="Diagnosis")
59
  ],
60
- title="__TITLE__ Detection"
61
  )
62
 
63
  interface.launch()
 
38
  output = results[0].plot()[:, :, ::-1] # BGR → RGB
39
 
40
  if len(boxes) > 0:
41
+ diagnosis = "⚠️ Swelling detected."
42
  else:
43
+ diagnosis = "🟢 No swelling detected."
44
 
45
  return [output, diagnosis]
46
 
 
50
  fn=detect,
51
  inputs=[
52
  gr.Image(type="pil", label="Upload Image"),
53
+ gr.Slider(0, 1, value=0.5, step=0.05, label="Confidence Threshold"),
54
  gr.Slider(0, 1, value=0.5, step=0.05, label="NMS IoU Threshold"),
55
  ],
56
  outputs=[
57
+ gr.Image(label="Swelling Detection Result"),
58
  gr.Textbox(label="Diagnosis")
59
  ],
60
+ title="Swelling Detection"
61
  )
62
 
63
  interface.launch()