moved resize to tf compose

app.py

@@ -35,6 +35,7 @@ model.eval()
 
 tf = v2.Compose([
     v2.ToImage(),
+    v2.Resize((64, 64)),
     v2.ToDtype(torch.float32, scale=True),
 ])
 
@@ -42,10 +43,11 @@ tf = v2.Compose([
 def predict(image):
     if image is None:
         return None
-
-    img_resized = image.resize((64, 64), Image.Resampling.LANCZOS)
-
-    img_tensor = tf(img_resized).unsqueeze(0).to(device)
+    
+    
+    pil_img = Image.fromarray(image.astype('uint8'), 'RGB')
+    
+    img_tensor = tf(pil_img).unsqueeze(0).to(device)
 
     with torch.no_grad():
         outputs = model(img_tensor)