Spaces:

prithivMLmods
/

PROMPT-EXTEND

Running

prithivMLmods commited on Jun 10, 2024

Commit

3a454bf

verified ·

1 Parent(s): e62241f

Update app.py

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='prithivMLmods/Stark-Prompt-Extender')
 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"]