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