Update app.py

app.py

@@ -101,7 +101,7 @@ def prediction(image_path):
         ax[1].imshow(image_np,aspect='auto');
 
 
-        return objects,image_np
+        return objects,ax[1]
     
     
         
@@ -129,7 +129,7 @@ def prediction(image_path):
     response = {"predictions": [result]}
     total_time_end = time.time()
     print("total time : ",round((total_time_end-total_time_start),2))
-    return simplejson.dumps(response),image_np
+    return simplejson.dumps(response),fig
 
 inputs = gr.inputs.Image(type = 'filepath')
 EXAMPLES = ["img1.jpg","img2.jpg","img6.jpg","img7.jpg","img8.jpg","img4.jpg","img10.jpg"]