Update app.py

app.py

@@ -11,7 +11,7 @@ from pathlib import Path
 
 transformer = transforms.Compose([    
     transforms.ToTensor(),                            
-   # transforms.Resize([224,224]),
+    transforms.Resize([640,640]),
 
 ])
 
@@ -70,7 +70,8 @@ model = Model(confg='./configs/rtdetr/rtdetr_r101vd_6x_coco.yml',ckpt="./checkpo
 
 
 def detect(img):
-  img = img.resize((640,640))
+  print(img)
+  #img = img.resize((640,640))
   t_img = transformer(img).unsqueeze(0)#.unsqueeze(0)      #[1,1,640,640]
   size = torch.tensor([[t_img.shape[2], t_img.shape[3]]])
   #print(t_img.shape)