Spaces:

daspartho
/

prompt-extend

Runtime error

daspartho commited on Nov 1, 2022

Commit

a8b3714

1 Parent(s): 7d17050

to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import pipeline
 import gradio as gr
-pipe = pipeline('text-generation', model='daspartho/prompt-extend', device=0)
 def extend_prompt(prompt):
     return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]

 from transformers import pipeline
 import gradio as gr
+pipe = pipeline('text-generation', model='daspartho/prompt-extend')
 def extend_prompt(prompt):
     return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]