Update app.py

app.py

@@ -8,9 +8,11 @@ API_TOKEN = st.secrets["HF_API_TOKEN"]
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 # Function to query the model
-def query(image_bytes):
-    # Prepare the payload with binary image data
-    files = {"inputs": image_bytes}
+def query(image):
+    # Prepare the payload with binary image data and filename
+    files = {
+        "inputs": (image.name, image, "image/png" if image.name.endswith(".png") else "image/jpeg")
+    }
     response = requests.post(API_URL, headers=headers, files=files)
     return response.json()
 
@@ -28,11 +30,11 @@ if uploaded_file is not None:
 
     st.write("Classifying...")
 
-    # Read the uploaded file as bytes
-    image_bytes = uploaded_file.read()  # Read the file as bytes
-
     # Send the image to the model
-    result = query(image_bytes)
+    result = query(uploaded_file)
+
+    # Debugging: Display the raw result from the Hugging Face API
+    st.write(result)  # This will display the full API response for debugging
 
     # Display the result
     if "error" in result: