Upload app.py with huggingface_hub

app.py

@@ -77,11 +77,30 @@ def _human_label(c):
         return CLASS_LABELS.get(c, str(c))
 
 # Do the prediction!
-def do_predict(pil_img: PIL.Image.Image):
-    # Make sure there's actually an image to work with
-    if pil_img is None:
+def do_predict(pil_img_input): # Renamed input parameter for clarity
+    # Check the type of the input
+    if pil_img_input is None:
         # Return empty results for the outputs if no image is provided
-        return {}, pandas.DataFrame(columns=["Predicted label", "Confidence (%)"])
+        return {"no stop sign": 0.0, "stop sign": 0.0}
+        
+    # Handle potential tuple input from webcam
+    if isinstance(pil_img_input, tuple):
+        print(f"Received tuple input from webcam: {pil_img_input}") # Print for debugging
+        # Assuming the tuple contains the image data as the first element
+        if pil_img_input and isinstance(pil_img_input[0], PIL.Image.Image):
+            pil_img = pil_img_input[0]
+            print("Successfully extracted PIL Image from tuple.")
+        else:
+            # If we can't extract a PIL Image from the tuple
+            print("Could not extract PIL Image from tuple input.")
+            return {"Error": "Could not process webcam input format."}
+    elif isinstance(pil_img_input, PIL.Image.Image):
+        pil_img = pil_img_input
+    else:
+        # Handle unexpected input types
+        print(f"Received unexpected input type: {type(pil_img_input)}")
+        return {"Error": f"Unexpected input format: {type(pil_img_input)}"}
+
 
     try:
         # IF we have something to work with, save it and prepare the input
@@ -114,7 +133,8 @@ def do_predict(pil_img: PIL.Image.Image):
     except Exception as e:
         # Return an error message or empty results if something goes wrong during processing
         print(f"Error processing image: {e}")
-        return {}, pandas.DataFrame(columns=["Predicted label", "Confidence (%)"])
+        # Return a dictionary with error information or default values
+        return {"Error": f"Processing failed: {e}"}
 
 
 # Representative example images! These can be local or links.