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()