Update app.py

app.py

@@ -48,25 +48,10 @@ transform = transforms.Compose([
 OBJECT_NAMES = ['enemies']
 
 
-def detect_objects_in_image(image):
-    # Ensure image is a PIL Image
-    data = Image.fromarray(image)
-
-    if isinstance(data, torch.Tensor):
-        print("hey")
-        image = transforms.ToPILImage()(data)  # Convert tensor to PIL image
-
-    if isinstance(data, Image.Image):
-        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))
-        print(type(data))
-        exit()
-
-    # Apply transformation
+ef detect_objects_in_image(image):
     img_tensor = transform(image).unsqueeze(0)
+    orig_w, orig_h = image.size
+
 
     with torch.no_grad():
         pred = model(img_tensor)[0]