Update app.py

app.py

@@ -101,7 +101,7 @@ def prediction(image_path):
         ax[1].imshow(image_np,aspect='auto');
 
 
-        return objects,image_np_orignal
+        return objects,image_np
     
     
         
@@ -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),fig
+    return simplejson.dumps(response),image_np
 
 inputs = gr.inputs.Image(type = 'filepath')
 EXAMPLES = ["img1.jpg","img2.jpg","img6.jpg","img7.jpg","img8.jpg","img4.jpg","img10.jpg"]
@@ -148,4 +148,5 @@ demo_app = gr.Interface(
     #live = True,
     theme = 'huggingface'
 )
-demo_app.launch()
+demo_app.launch()
+