Update app.py

app.py

@@ -30,14 +30,19 @@ def image_preprocess(image):
 model = YOLO('best (1).pt')
 
 def response(image):
-    results = model(image)
-    for i, r in enumerate(results):
+  print(image)
+  results = model(image)
+  for i, r in enumerate(results):
     # Plot results image
       im_bgr = r.plot()  # BGR-order numpy array
       im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
     
     # im_rgb = Image.fromarray(im_rgb)
-    return im_rgb
+      
+      return im_rgb
+
+
+
 
 iface = gr.Interface(fn=response, inputs="image", outputs="image")
 iface.launch()