Wootang01's picture
Update app.py
c6c42e2
raw
history blame
165 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-generation", model='EleutherAI/pythia-800m')
gr.Interface.from_pipeline(pipe).launch()