Create app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+import requests
+import os
+
+API_URL = "https://api-inference.huggingface.co/models/gabehubner/trained-distilbert-model"
+headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+
+def classify_text(text):
+    response = requests.post(API_URL, headers=headers, json={"inputs": text})
+    if response.status_code != 200:
+        return f"Error: {response.text}"
+
+    results = response.json()
+    return f"Label: {results[0]['label']} (Confidence: {results[0]['score']:.2f})"
+
+interface = gr.Interface(
+    fn=classify_text,
+    inputs=gr.Textbox(placeholder="Enter text here..."),
+    outputs="text",
+    title="Sentiment Classifier",
+    description="Enter text and see whether it's classified as positive or negative!",
+)
+
+if __name__ == "__main__":
+    interface.launch()