Epstein-LLM / app.py
sleeping4cat's picture
Create app.py
8000cac verified
raw
history blame contribute delete
297 Bytes
import gradio as gr
from transformers import pipeline
generator = pipeline("text-generation", model="sleeping-ai/Epstein-LLM")
def generate_text(prompt):
return generator(prompt, max_new_tokens=200)[0]['generated_text']
gr.Interface(fn=generate_text, inputs="text", outputs="text").launch()