Spaces:
Sleeping
Sleeping
import gradio as gr | |
import requests | |
API_URL = "https://api-inference.huggingface.co/models/electric-otter/litemachine-quick-4" | |
headers = {"Authorization": "Bearer YOUR_HUGGINGFACE_API_TOKEN"} | |
def generate_text(input_text): | |
payload = { | |
"inputs": input_text, | |
"parameters": {"max_length": 100, "num_return_sequences": 1}, | |
} | |
response = requests.post(API_URL, headers=headers, json=payload) | |
return response.json()[0]['generated_text'] | |
interface = gr.Interface(fn=generate_text, inputs="text", outputs="text") | |
interface.launch() | |