kevinconka commited on
Commit
8cdef7f
·
1 Parent(s): 941fa10

Refactor inference function in app.py to remove image file path from output and update UI components accordingly.

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -57,7 +57,7 @@ def inference(image_fp):
57
  """Run inference on image and return annotated image."""
58
  image = imread(image_fp)
59
  results = model(image)
60
- return results.draw(image), image_fp
61
 
62
 
63
  def flag_img_input(
@@ -98,14 +98,13 @@ with gr.Blocks(theme=theme, css=css, title="SEA.AI Vision Demo") as demo:
98
  submit = gr.Button("Submit", variant="primary")
99
  with gr.Column():
100
  img_output = gr.Image(label="output", interactive=False)
101
- img_filepath = gr.Textbox(label="image_filepath", visible=False)
102
  flag = gr.Button(FLAG_TXT, visible=False)
103
  notice = gr.Markdown(value=NOTICE, visible=False)
104
 
105
  examples = gr.Examples(
106
  examples=glob.glob("examples/*.jpg"),
107
  inputs=img_input,
108
- outputs=[img_output, img_filepath],
109
  fn=inference,
110
  cache_examples=True,
111
  )
@@ -118,7 +117,7 @@ with gr.Blocks(theme=theme, css=css, title="SEA.AI Vision Demo") as demo:
118
  submit.click(check_image, [img_input], None, show_api=False).success(
119
  inference,
120
  [img_input],
121
- [img_output, img_filepath],
122
  api_name="inference",
123
  )
124
 
 
57
  """Run inference on image and return annotated image."""
58
  image = imread(image_fp)
59
  results = model(image)
60
+ return results.draw(image)
61
 
62
 
63
  def flag_img_input(
 
98
  submit = gr.Button("Submit", variant="primary")
99
  with gr.Column():
100
  img_output = gr.Image(label="output", interactive=False)
 
101
  flag = gr.Button(FLAG_TXT, visible=False)
102
  notice = gr.Markdown(value=NOTICE, visible=False)
103
 
104
  examples = gr.Examples(
105
  examples=glob.glob("examples/*.jpg"),
106
  inputs=img_input,
107
+ outputs=img_output,
108
  fn=inference,
109
  cache_examples=True,
110
  )
 
117
  submit.click(check_image, [img_input], None, show_api=False).success(
118
  inference,
119
  [img_input],
120
+ img_output,
121
  api_name="inference",
122
  )
123