EmmaL1 commited on
Commit
5f39df4
·
verified ·
1 Parent(s): 3b913aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,11 +14,11 @@ def img2text(img):
14
  def text2story(text):
15
  text_generation_model = pipeline("text-generation",
16
  model="meta-llama/Meta-Llama-3-8B")
17
- story_text = text_generation_model(text)[0]["generated_text"]
18
  generated_story = text_generation_model(story_text,
19
  max_length=100,
20
  num_return_sequences=1)
21
- return generated_story
22
 
23
  # text2audio
24
  def text2audio(story_text):
 
14
  def text2story(text):
15
  text_generation_model = pipeline("text-generation",
16
  model="meta-llama/Meta-Llama-3-8B")
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,
20
  num_return_sequences=1)
21
+ return generated_story[0]['generated_text']
22
 
23
  # text2audio
24
  def text2audio(story_text):