Update app.py

app.py

@@ -275,11 +275,13 @@ def predict(image):
     annotated_img = output[0]
     # annotated_img = (output[0] / 255.0 - mean)/std
     # annotated_img = classes[output[0][0].argmax(0)]
-    print("type annotated image:", type(annotated_img))
-    print(annotated_img)
+    print("Annotated image type:", type(annotated_img))
+    print("Annotated image:", annotated_img)
+    print("Min value of image:", np.min(annotated_img))
+    print("Max value of image:", np.max(annotated_img))
 
     # Convert to PIL Image
-    annotated_img = Image.fromarray(annotated_img)
+    # annotated_img = Image.fromarray(annotated_img)
 
     return annotated_img
     
@@ -287,8 +289,8 @@ def predict(image):
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(sources=["webcam"], type="numpy"),  # Accepts image input
-    # outputs="annotatedimage"  # Customize based on your output format
-    outputs=gr.Image(type="pil"),  # Accepts image input
+    outputs="image"  # Customize based on your output format
+    # outputs=gr.Image(type="pil"),  # Accepts image input
 )
 
 if __name__ == "__main__":