Charleshhh commited on
Commit
dbe8f63
·
verified ·
1 Parent(s): 45e1ffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def img2text(url):
12
 
13
  # text2story
14
  def text2story(text):
15
- story_generator = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
16
  story = story_generator(text, max_length=200, num_return_sequences=1)[0]['generated_text']
17
  return story
18
 
 
12
 
13
  # text2story
14
  def text2story(text):
15
+ story_generator = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
16
  story = story_generator(text, max_length=200, num_return_sequences=1)[0]['generated_text']
17
  return story
18