Update app.py

app.py

@@ -32,15 +32,20 @@ model.eval()
 # model_path = "./last.pt"
 # model = torch.jit.load(model_path, map_location=torch.device("cpu"))
 # model.eval()
-
-transform = transforms.Compose([
-    transforms.Resize((640, 640)),
-    transforms.ToTensor(),
+transform=transforms.Compose([
+    transforms.ToPILImage(),
+    transforms.Resize((512,640)),
+    transforms.ToTensor()
 ])
+# transform = transforms.Compose([
+#     transforms.Resize((640, 640)),
+#     transforms.ToTensor(),
+# ])
 
 OBJECT_NAMES = ['enemies']
 
 def detect_objects_in_image(image):
+    
     img_tensor = transform(image).unsqueeze(0)
     orig_w, orig_h = image.size