Update app.py

app.py

@@ -301,7 +301,10 @@ def predict(image):
     print("annotated_img shape after normalization:", annotated_img.shape)
 
     # Convert to PIL Image
-    # annotated_img = Image.fromarray(annotated_img)
+    annotated_img = Image.fromarray(annotated_img)
+
+    print("PIL Image type:", type(annotated_img))
+    # print("PIL Image shape:", annotated_img.shape)
 
     return annotated_img
     
@@ -309,8 +312,8 @@ def predict(image):
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(sources=["webcam"], type="numpy"),  # Accepts image input
-    outputs="image"  # Customize based on your output format
-    # outputs=gr.Image(type="numpy"),  # Accepts image input
+    # outputs="image"  # Customize based on your output format
+    outputs=gr.Image(type="pil"),  # Accepts image input
 )
 
 if __name__ == "__main__":