Update app.py
Browse files
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=[
|
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()
|