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