|
import gradio as gr |
|
import requests |
|
import json |
|
|
|
|
|
|
|
def query_huggingface_api(prompt, api_url, api_key): headers = {"Authorization": f"Bearer {api_key}"} payload = {"inputs": prompt} response = requests.post(api_url, headers=headers, json=payload) |
|
|
|
if response.status_code == 200: |
|
return response.json()[0]['generated_text'] |
|
else: |
|
return f"Error: API request failed with status code {response.status_code}" |
|
|
|
Define the Gradio interface |
|
|
|
def chat_with_model(user_input): API_URL = "https://api-inference.huggingface.co/models/YOUR_MODEL_NAME" API_KEY = "YOUR_HF_API_KEY" return query_huggingface_api(user_input, API_URL, API_KEY) |
|
|
|
iface = gr.Interface(fn=chat_with_model, inputs="text", outputs="text", title="ChatDev AI") |
|
|
|
Launch the Gradio app |
|
|
|
if name == "main": iface.launch() |
|
|
|
|
