gherget commited on
Commit
6deaf33
1 Parent(s): 81a4b09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -91,7 +91,7 @@ def inference(img,unique_only):
91
 
92
  outputs = predictor(im)
93
  v = Visualizer(im[:, :, ::-1], metadata)
94
- out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
95
 
96
  detected_objects = []
97
  object_list_str = []
@@ -125,7 +125,7 @@ def inference(img,unique_only):
125
  output_str += line + "\n"
126
 
127
  return (
128
- Image.fromarray(np.uint8(out.get_image())).convert("RGB"),
129
  detected_objects
130
  )
131
 
@@ -144,6 +144,6 @@ with gr.Blocks() as demo:
144
  output_desc = gr.Textbox(label="Description for using in ChatGPT", lines=5)
145
  # outputjson = gr.JSON(label="Detected Objects")
146
 
147
- btn_detic.click(fn=inference, inputs=[inp,chk], outputs=[outviz, output_desc], api_name="detect")
148
 
149
  demo.launch()
 
91
 
92
  outputs = predictor(im)
93
  v = Visualizer(im[:, :, ::-1], metadata)
94
+ #out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
95
 
96
  detected_objects = []
97
  object_list_str = []
 
125
  output_str += line + "\n"
126
 
127
  return (
128
+ #Image.fromarray(np.uint8(out.get_image())).convert("RGB"),
129
  detected_objects
130
  )
131
 
 
144
  output_desc = gr.Textbox(label="Description for using in ChatGPT", lines=5)
145
  # outputjson = gr.JSON(label="Detected Objects")
146
 
147
+ btn_detic.click(fn=inference, inputs=[inp,chk], outputs=[output_desc], api_name="detect")
148
 
149
  demo.launch()