EmmaL1 commited on
Commit
1eebd1b
·
verified ·
1 Parent(s): 5af1e9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def img2text(img):
13
  # text2story
14
  def text2story(text):
15
  text_generation_model = pipeline("text-generation",
16
- model="Qwen/Qwen2.5-1.5B-Instruct")
17
  story_text = f"Once upon a time in a land far, far away, {text}"
18
  generated_story = text_generation_model(story_text,
19
  max_length=100,
 
13
  # text2story
14
  def text2story(text):
15
  text_generation_model = pipeline("text-generation",
16
+ model="openai-community/gpt2")
17
  story_text = f"Once upon a time in a land far, far away, {text}"
18
  generated_story = text_generation_model(story_text,
19
  max_length=100,