Update app.py

app.py

@@ -10,9 +10,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"<BOS> {text}"
-    generated = story_generator(prompt, max_new_tokens=140, do_sample=True, temperature=0.7)
+    generated = story_generator(prompt, max_new_tokens=150, do_sample=True, temperature=0.7)
     story_text = generated[0]['generated_text']
 
+    # Remove leading ". " if present
+    if story_text.startswith(". "):
+        story_text = story_text[2:]
+    
     # Split into sentences
     sentences = re.split(r'(?<=[.!?])\s+', story_text.strip())
     
@@ -39,7 +43,7 @@ def text2story(text):
     return final_story
 
 def text2audio(story_text):
-    audio_generator = pipeline("text-to-speech", model="ylacombe/vits_vctk_scottish_female")
+    audio_generator = pipeline("text-to-speech", model="khof312/mms-tts-swh-female-1")
     audio_output = audio_generator(story_text)
     return audio_output