Fanny1366 commited on
Commit
c27417e
·
verified ·
1 Parent(s): 30c1175

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="Qwen/QwQ-32B-Preview")
18
  story_text = story_pipeline(text, max_length=200)[0]['generated_text']
19
  return story_text
20
 
 
14
 
15
  # text2story
16
  def text2story(text):
17
+ story_pipeline = pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
18
  story_text = story_pipeline(text, max_length=200)[0]['generated_text']
19
  return story_text
20