Update app.py

app.py

@@ -11,10 +11,7 @@ n_raw_model = YOLO("N-Raw.pt")
 s_raw_model = YOLO("S-Raw.pt")
 
 def snap(image, model, conf, iou):
-    # Convert the image to a numpy array
-    print(type(image))
-    image = np.array(image)
-
+    
     # Run the selected model
     results = None
     if model == "M-Raw":
@@ -48,7 +45,7 @@ def snap(image, model, conf, iou):
         cv2.putText(image, f"{classes} {probs:.2f}", (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
 
     # Convert the resulting image to a PIL image
-    resulting_image = Image.fromarray(image)
+    resulting_image = image
 
     # Get the labels
     # labels = results.pandas().xyxy[0]["name"].values