Spaces:
Sleeping
Sleeping
File size: 558 Bytes
01f639a dc72004 01f639a dc72004 01f639a dc72004 01f639a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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()
|