Spaces:
Running
Running
Update app.py
Browse files
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="
|
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="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
|
16 |
}
|
17 |
return models
|
18 |
|