Update app.py

app.py

@@ -6,6 +6,8 @@ from huggingface_hub import snapshot_download
 import os
 import cv2
 
+newsize = (640, 640)
+
 model_path = "best_int8_openvino_model"
 
 # Example paths for Gradio
@@ -23,7 +25,8 @@ def load_model(repo_id):
 
 # Predict the image
 def predict(pilimg):
-    source = cv2.resize(pilimg, (640,640))
+    #source = pilimg
+    source = pilimg.resize(newsize)
     # x = np.asarray(pilimg)
     # print(x.shape)