Update app.py

app.py

@@ -85,17 +85,29 @@ if st.session_state['generated']:
             message(st.session_state["generated"][i], key=str(i), avatar_style="fun-emoji")
 
 
-API_URL = "https://api-inference.huggingface.co/models/Prasanna18/indian-food-classification"
-HEADERS = {"key"}
+import requests
+import streamlit as st
+import time
 
-def query_image_classification(image_bytes):
-    try:
-        response = requests.post(API_URL, headers=HEADERS, data=image_bytes)
-        result = response.json()
-        return result
-    except Exception as e:
-        st.error(f"An error occurred during image classification: {str(e)}")
-        return None
+API_URL = "https://api-inference.huggingface.co/models/Prasanna18/indian-food-classification"
+HEADERS = {"Authorization": "Bearer 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"}
+
+def query_image_classification(image_bytes, max_retries=3):
+    for retry in range(max_retries):
+        try:
+            response = requests.post(API_URL, headers=HEADERS, data=image_bytes)
+            response.raise_for_status()  # Raise an error for non-2xx HTTP responses
+            result = response.json()
+            return result
+        except requests.exceptions.RequestException as e:
+            st.error(f"An error occurred during the API request: {str(e)}")
+        except ValueError as e:
+            st.error(f"An error occurred while processing the API response: {str(e)}")
+        if retry < max_retries - 1:
+            st.warning(f"Retrying request (attempt {retry + 1}/{max_retries})...")
+            time.sleep(1)  # Wait for a moment before retrying
+    st.error("No classification result received after multiple retries.")
+    return None
 
 if image_upload:
     image_bytes = image_upload.read()
@@ -116,8 +128,6 @@ if image_upload:
                 st.error("Invalid classification result format. Missing 'label' key.")
         else:
             st.error("Invalid classification result format or empty result list.")
-    else:
-        st.error("No classification result received.")
 
 
 import pydeck as pdk