CR7CAD commited on
Commit
63de7ec
·
verified ·
1 Parent(s): b2da4f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load_models():
12
  """Load all models and cache them for faster execution"""
13
  models = {
14
  "image_captioner": pipeline("image-to-text", model="sooh-j/blip-image-captioning-base"),
15
- "story_generator": pipeline("text-generation", model="Qwen/Qwen2.5-Coder-0.5B-Instruct-GGUF")
16
  }
17
  return models
18
 
 
12
  """Load all models and cache them for faster execution"""
13
  models = {
14
  "image_captioner": pipeline("image-to-text", model="sooh-j/blip-image-captioning-base"),
15
+ "story_generator": pipeline("text-generation", model="Mihaiii/Pallas-0.5")
16
  }
17
  return models
18