EmmaL1 commited on
Commit
7f32a8e
·
verified ·
1 Parent(s): 932211d

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="meta-llama/Meta-Llama-3-8B")
17
  story_text = "Once upon a time in a land far, far away"
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="distilbert/distilgpt2")
17
  story_text = "Once upon a time in a land far, far away"
18
  generated_story = text_generation_model(story_text,
19
  max_length=100,