Update app.py

app.py

@@ -21,18 +21,15 @@ 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
-    
-    # # Make prediction
-    # with torch.no_grad():
-    #     output = model(img_tensor)
-    
     # Process output (adjust based on your model's format)
     results = model(image)
     print("ran the model")
     annotated_img = results[0].plot()
     print("got annotated img")
     print("type annotated img:", type(annotated_img))
+
+    annotated_img = cv2.cvtColor(annotated_img, cv2.COLOR_RGB2BGR)
+    print("converted the colour to BGR.")
     
     return annotated_img