Spaces:

Wootang01
/

text_generator_four

Build error

Wootang01 commited on Dec 25, 2022

Commit

8cbac19

1 Parent(s): 07711c2

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/gpt-neo-2.7B')
 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='Wootang01/distilgpt2-finetuned-hkdse-english-paper4')
 def generate(text):
     result = generator(text, max_length=100, num_return_sequences=1)