Update app.py

app.py

@@ -1,7 +1,40 @@
 import gradio as gr
+import requests
 
-def greet(name):
-    return "Hello " + name + "!!"
+def query_huggingface_api(api_key, model, prompt):
+    API_URL = f"https://api-inference.huggingface.co/models/{model}"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "inputs": prompt
+    }
+
+    response = requests.post(API_URL, headers=headers, json=data)
+    result = response.json()
+
+    # Check for errors or handle the response appropriately.
+    if isinstance(result, dict) and 'error' in result:
+        return f"Error: {result['error']}"
+
+    return result['generated_text'] if isinstance(result, dict) and 'generated_text' in result else result
+
+# List of some available models
+model_list = [
+    "stabilityai/stable-diffusion-xl-base-1.0",
+]
+
+iface = gr.Interface(
+    fn=query_huggingface_api,
+    inputs=[
+        gr.Textbox(label="Hugging Face API Key", placeholder="Enter your Hugging Face API Key here..."),
+        gr.Dropdown(choices=model_list, label="Model"),
+        gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt")
+    ],
+    outputs="text",
+    title="Hugging Face Model API Explorer",
+    description="Enter your API Key, select a model, and enter a prompt to generate text using the Hugging Face Inference API."
+)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()