Update app.py

app.py

@@ -8,13 +8,13 @@ def img2text(url):
 
 def text2story(text):
     story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", device_map="auto", return_full_text=False)
-    prompt = f"Under 100 words about: {text}."
-    generated = story_generator(text, max_new_tokens=120, top_k=120, do_sample=True, temperature=0.7)
+    prompt = f"A short story under 100 words about: {text}."
+    generated = story_generator(text, max_length=100, max_new_tokens=120, do_sample=True, temperature=0.7, num_return_sequences=1)
     story_text = generated[0]['generated_text']
     return story_text
 
 def text2audio(story_text):
-    audio_generator = pipeline("text-to-speech", model="Nick256/tts-tacotron2-commonvoice-single-female")
+    audio_generator = pipeline("text-to-speech", model="ylacombe/vits_vctk_scottish_female")
     audio_output = audio_generator(story_text)
     return audio_output