Spaces:

syedfaisalabrar
/

Mouse-Feet-Detection

Sleeping

syedfaisalabrar commited on Jan 16

Commit

3adf147

verified ·

1 Parent(s): 4924459

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ model = YOLO('mouseFeetDetection.pt')
 # Define the prediction function for YOLO model
 def predict(image):
-        if image is None:
-            return None
     image = Image.fromarray(np.array(image))
     results = model(image)
     annotated_image = results[0].plot()

 # Define the prediction function for YOLO model
 def predict(image):
+    if image is None:
+        return None
     image = Image.fromarray(np.array(image))
     results = model(image)
     annotated_image = results[0].plot()