Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import yolov7
 import tempfile
 import cv2
+import numpy as np
 from pathlib import Path
 
 def image_fn(image, model_path, image_size, conf_threshold, iou_threshold):
@@ -20,11 +21,19 @@ def image_fn(image, model_path, image_size, conf_threshold, iou_threshold):
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)
-    img = results.render()[0]
-    img.flags.writeable = True  # Make the image writable
+    
+    # Convert PIL image to NumPy array if necessary
+    if isinstance(results.render()[0], np.ndarray):
+        img = results.render()[0]
+    else:
+        img = np.array(results.render()[0])
+    
+    # Ensure the image array is writable
+    img.flags.writeable = True
+    
     return img
 
-# Use gr.components instead of gr.inputs and gr.outputs
+# Use gr instead of gr.inputs and gr.outputs
 image_interface = gr.Interface(
     fn=image_fn,
     inputs=[