litemachine / app.py
electric-otter's picture
Update app.py
dc72004 verified
raw
history blame contribute delete
558 Bytes
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()