demo / app.py
galnikolai's picture
Update app.py
825d2ce verified
raw
history blame contribute delete
295 Bytes
from transformers import pipeline
import gradio as gr
generator = pipeline("text-generation", model="Markeaze/dashboard")
def generate_response(prompt):
return generator(prompt, max_length=50)[0]["generated_text"]
gr.Interface(fn=generate_response, inputs="text", outputs="text").launch()