Fanny1366 commited on
Commit
42facc3
·
verified ·
1 Parent(s): ef7790c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def img2text(url):
14
 
15
  # text2story
16
  def text2story(text):
17
- story_pipeline = pipeline("text-generation", model="qihoo360/TinyR1-32B-Preview")
18
  result = story_pipeline(text, max_length=200)
19
  story_text = result[0]['generated_text']
20
  return story_text
 
14
 
15
  # text2story
16
  def text2story(text):
17
+ story_pipeline = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
18
  result = story_pipeline(text, max_length=200)
19
  story_text = result[0]['generated_text']
20
  return story_text