Update app.py

app.py

@@ -224,15 +224,15 @@ def predict(image):
     # Preprocess the image
     img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
     
-    # Make prediction
-    with torch.no_grad():
-        output = model(img_tensor)
+    # # Make prediction
+    # with torch.no_grad():
+    #     output = model(img_tensor)
     
     # Process output (adjust based on your model's format)
     # return output  # or post-process the results as needed
     results = model(image)
-    print(type(results))
-    print(results)
+    # print(type(results))
+    # print(results)
     annotated_img = results[0].plot()
     return annotated_img