Spaces:

Wootang01
/

text_generator_two

Runtime error

Wootang01 commited on Dec 27, 2022

Commit

f2eb06e

1 Parent(s): f50f4fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-generator = pipeline('text-generation', model='EleutherAI/pythia-350m-deduped')
 def generate(text):
     result = generator(text, max_length=100, num_return_sequences=1)

 import gradio as gr
 from transformers import pipeline
+generator = pipeline('text-generation', model='EleutherAI/pythia-800m')
 def generate(text):
     result = generator(text, max_length=100, num_return_sequences=1)