Update app.py

app.py

@@ -243,7 +243,7 @@ transform = T.Compose([
     T.ToTensor(),
 ])
 
-def predict(image):
+    def predict(image):
     # Preprocess the image
     img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
     
@@ -254,6 +254,7 @@ def predict(image):
     # Process output (adjust based on your model's format)
     # return output  # or post-process the results as needed
     results = model(image)
+    print(type(results))
     annotated_img = results.plot()
     return annotated_img