Update app.py

app.py

@@ -35,8 +35,7 @@ model.eval()
 #     transforms.Resize((512,640)),
 #     transforms.ToTensor()
 # ])
-transform = transforms.Compose([
-    transforms.ToPILImage(),  # Ensure input is a PIL image
+transform = transforms.Compose([  # Ensure input is a PIL image
     transforms.Resize((512, 640)),
     transforms.ToTensor()
 ])
@@ -49,20 +48,24 @@ OBJECT_NAMES = ['enemies']
 
 
 def detect_objects_in_image(image):
+    
     print(type(image))
     print(np.ndarray.view(image))
-    img_tensor = transform(image).unsqueeze(0)
+    
+    
     print(image.size)
     if isinstance(image, np.ndarray):
         print("Converting NumPy array to PIL Image")
         image = Image.fromarray(image)
     print(image.size)
+    img_tensor = transform(image).unsqueeze(0)
     orig_w, orig_h = image.size
     print("passed1")
 
-
+    print(torch.no_grad())
     with torch.no_grad():
         pred = model(img_tensor)[0]
+    
     print("Passed2")
 
     if isinstance(pred[0], torch.Tensor):