Update app.py

app.py

@@ -48,10 +48,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")  # Use GPU
 model = attempt_load(model_path, device=device)  # Placeholder for model loading
 model.eval()  # Set the model to evaluation mode
 
-def preprocess_image(image_path):
-#def preprocess_image(image):
-    img0 = cv2.imread(image_path)
-    #img0 = letterbox(image, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
+#def preprocess_image(image_path):
+def preprocess_image(image):
+    #img0 = cv2.imread(image_path)
+    img0 = letterbox(image, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
     img = letterbox(img0, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
     print("in preprocess:",img.shape)
     img = img.transpose(2, 0, 1)[::-1]  # Convert BGR to RGB, to 3x416x416
@@ -222,4 +222,3 @@ def build_interface():
 if __name__ == '__main__':
     demo = build_interface()
     demo.launch
-'''