Update app.py

app.py

@@ -35,8 +35,9 @@ def plot_img_bbox(img, target):
             cv2.putText(img, label, (xmin, ymin - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 0, 255), 2)
 
         # Display the image with detections
-        filename = 'pred.jpg'
-        cv2.imwrite(filename, img)
+        #filename = 'pred.jpg'
+        #cv2.imwrite(filename, img)
+        return img
 
 # transform image
 test_transforms = A.Compose([
@@ -71,15 +72,18 @@ def predict(img) :
 
     nms_prediction = apply_nms(predictions, iou_thresh=0.1)
 
-    plot_img_bbox(np.array(img), nms_prediction)
+    pred = plot_img_bbox(np.array(img), nms_prediction)
 
-    pred = np.array(Image.open("pred.jpg"))
+    #pred = np.array(Image.open("pred.jpg"))
     
     # Calculate the prediction time
     pred_time = round(timer() - start_time, 5)
     
     # Return the prediction dictionary and prediction time 
-    return pred,pred_time
+    return pred
+    
+image = gr.components.Image()
+out_im = gr.components.Image()
 
 ### 4. Gradio app ###
 # Create title, description and article strings
@@ -89,13 +93,13 @@ article = "Created by data354."
 
 # Create examples list from "examples/" directory
 example_list = [["examples/" + example] for example in os.listdir("examples")]
-
+#[gr.Label(label="Predictions"), # what are the outputs?
+#gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
+# Create examples list from "examples/" directory
 # Create the Gradio demo
 demo = gr.Interface(fn=predict, # mapping function from input to output
                     inputs=gr.Image(type="pil"), # what are the inputs?
-                    outputs=[gr.Label(label="Predictions"), # what are the outputs?
-                             gr.Number(label="Prediction time (s)")], # our fn has two outputs, therefore we have two outputs
-                    # Create examples list from "examples/" directory
+                    outputs=out_im
                     examples=example_list, 
                     title=title,
                     description=description,