Update app.py

app.py

@@ -21,7 +21,7 @@ def predict(image):
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     print("converted the colour to RGB.")
     
-    img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
+    # img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
     
     # # Make prediction
     # with torch.no_grad():
@@ -29,9 +29,10 @@ def predict(image):
     
     # Process output (adjust based on your model's format)
     results = model(image)
+    print("ran the model")
     annotated_img = results[0].plot()
-
-    print(type(annotated_img))
+    print("got annotated img")
+    print("type annotated img:", type(annotated_img))
     
     return annotated_img