Update app.py

app.py

@@ -51,10 +51,11 @@ 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
 
-    if isinstance(data, Image.Image):
+    if isinstance(image, Image.Image):
         orig_w, orig_h = image.size  # PIL image size returns (width, height)
     else:
         raise TypeError(f"Expected a PIL Image but got ")