xujinheng666 commited on
Commit
109ddb8
·
verified ·
1 Parent(s): 42f31fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  def load_models():
9
  return {
10
  "image_to_text": pipeline("image-to-text", model="Salesforce/blip-image-captioning-base"),
11
- "story_gen": pipeline("text-generation", model="Qwen/Qwen2.5-1.5B-Instruct"),
12
  "text_to_speech": pipeline("text-to-speech", model="facebook/mms-tts-eng")
13
  }
14
 
 
8
  def load_models():
9
  return {
10
  "image_to_text": pipeline("image-to-text", model="Salesforce/blip-image-captioning-base"),
11
+ "story_gen": pipeline("text-generation", model="distilbert/distilgpt2"),
12
  "text_to_speech": pipeline("text-to-speech", model="facebook/mms-tts-eng")
13
  }
14