Wootang01's picture
Update app.py
4c476ba
raw
history blame
277 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline('text-generation', model="EleutherAI/pythia-125m")
description_ = "Kindly input text, submit, and the computer will predict subsequent words for your text input."
gr.Interface.from_pipeline(pipe).launch()