Update app.py

app.py

@@ -96,12 +96,7 @@ def inferenceScore(image_url, image, min_score, model_name):
 
     v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
     out = v.draw_instance_predictions(outputs["instances"].scores[1])
-
-    damage_class_map= {0:'damage'}
-    damage_prediction_classes = [ damage_class_map[el] + "_" + str(indx) for indx,el in enumerate(outputs["instances"].pred_classes.tolist())]
-    damage_polygon_centers = outputs["instances"].pred_boxes.get_centers().tolist()
-    damage_dict = dict(zip(damage_prediction_classes,damage_polygon_centers))   
-    return damage_dict
+    return out.item()
 
 title = "# DBMDZ Detectron2 Model Demo"
 description = """
@@ -130,12 +125,12 @@ with gr.Blocks() as demo:
 
     output_image = gr.Image(type="pil", label="Output")
 
-    label = gr.outputs.Label()
+    label = gr.Number
     inference_button = gr.Button("Submit")
     
     #inference_button.click(fn=inferenceScore, inputs=[url_input, image_input, min_score, model_name], outputs=output_image)
 
-    inference_button.click(fn=inferenceScore, inputs=[url_input, image_input, min_score, model_name], outputs=label)
+    inference_button.click(fn=inferenceScore, inputs=[url_input, image_input, min_score, model_name],  "number")
     
 
     gr.Markdown(footer)