Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def generate_caption(image_file):
|
|
13 |
return caption
|
14 |
|
15 |
def generate_story(caption):
|
16 |
-
story_generator = pipeline("text-generation", model="
|
17 |
messages = [
|
18 |
{
|
19 |
"role": "user",
|
|
|
13 |
return caption
|
14 |
|
15 |
def generate_story(caption):
|
16 |
+
story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-14B-Instruct-1M")
|
17 |
messages = [
|
18 |
{
|
19 |
"role": "user",
|