Update app.py

app.py

@@ -56,10 +56,11 @@ def inference(img):
             ax.text(x, y, "{} {:.0f}%".format(classes_to_labels[classes[idx] - 1], confidences[idx]*100), bbox=dict(facecolor='white', alpha=0.5))
     plt.axis('off')
     plt.draw()
-    return plt
+    plt.savefig("test.png",bbox_inches='tight')
+    return "test.png"
 
 inputs = gr.inputs.Image(type='file', label="Original Image")
-outputs = gr.outputs.Image(type="plot", label="Output Image")
+outputs = gr.outputs.Image(type="file", label="Output Image")
 
 title = "Single Shot MultiBox Detector model for object detection"
 description = "Gradio demo for Single Shot MultiBox Detector model for object detection by Nvidia. To use it upload an image or click an example images images. Read more at the links below"