Spaces:

Wootang01
/

text_generator_three

Configuration error

Wootang01 commited on Dec 25, 2022

Commit

7e000e8

1 Parent(s): 07e878c

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