Update app.py

app.py

@@ -50,13 +50,15 @@ OBJECT_NAMES = ['enemies']
 
 def detect_objects_in_image(image):
     # Ensure image is a PIL Image
-    image = Image.fromarray(image)
+    data = Image.fromarray(image)
 
     if isinstance(image, torch.Tensor):
-        image = transforms.ToPILImage()(image)  # Convert tensor to PIL image
+        print("hey")
+        image = transforms.ToPILImage()(data)  # Convert tensor to PIL image
 
     if isinstance(image, Image.Image):
-        orig_w, orig_h = image.size  # PIL image size returns (width, height)
+        print("hey2")
+        orig_w, orig_h = data.size  # PIL image size returns (width, height)
     else:
         raise TypeError(f"Expected a PIL Image but got ")
         print(type(image))