Spaces:

SamDaLamb
/

ValorantTracker

Running

SamDaLamb commited on Feb 12

Commit

de91a9d

verified ·

1 Parent(s): 1fb0186

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ OBJECT_NAMES = ['enemies']
 def detect_objects_in_image(image):
     # Ensure image is a PIL Image
-    data = Image.fromarray(image)
     if isinstance(image, torch.Tensor):
         image = transforms.ToPILImage()(image)  # Convert tensor to PIL image

 def detect_objects_in_image(image):
     # Ensure image is a PIL Image
+    image = Image.fromarray(image)
     if isinstance(image, torch.Tensor):
         image = transforms.ToPILImage()(image)  # Convert tensor to PIL image