Update app.py

app.py

@@ -2,22 +2,39 @@ import gradio as gr
 import requests
 import os
 
-API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-small"
+# Hugging Face Inference API endpoint (supported free model)
+API_URL = "https://api-inference.huggingface.co/models/t5-small"
+
+# Use your HF token (store in Space secrets as HF_TOKEN)
 HF_TOKEN = os.environ.get("HF_TOKEN")
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def chord_api(message: str, history: list[tuple[str,str]]):
+def chord_api(message: str, history: list[tuple[str, str]]):
+    """
+    Calls Hugging Face Inference API (t5-small) to classify chords.
+    """
     prompt = f"What chord do these notes form? {message}"
-    payload = {"inputs": prompt}
-    response = requests.post(API_URL, headers=headers, json=payload)
+    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+
     if response.status_code == 200:
-        return response.json()[0]["generated_text"]
+        try:
+            return response.json()[0]["generated_text"]
+        except Exception:
+            return f"⚠️ Unexpected response: {response.text}"
+
+    elif response.status_code == 503:
+        return "⏳ Model is loading, please try again in a moment."
+
     return f"❌ Error {response.status_code}: {response.text}"
 
+# Gradio chat interface
 chatbot = gr.ChatInterface(
     fn=chord_api,
     title="🎶 Chord Bot (API Version)",
-    description="This version calls Hugging Face’s Inference API (flan-t5-small)."
+    description=(
+        "This version queries Hugging Face’s free Inference API (`t5-small`). "
+        "Enter notes like `C E G` to see what chord they form."
+    ),
 )
 
 if __name__ == "__main__":