Update app.py

app.py

@@ -6,7 +6,8 @@ import gradio as gr
 def test_api_connection(query: str, api_key: str) -> str:
     """Tests the Hugging Face Inference API connection with the provided API key."""
     try:
-        client = InferenceClient(model="distilbert-base-uncased-finetuned-sst-2-english", token=api_key)
+        # Using a different model for testing
+        client = InferenceClient(model="distilbert-base-uncased", token=api_key)
         response = client.text_classification(query)
         return f"API connection successful! Response: {response}"
     except Exception as e: