Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -383,7 +383,7 @@ from TTS.api import TTS
|
|
383 |
|
384 |
# --------------- Lightweight Models ---------------
|
385 |
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
386 |
-
story_gen = pipeline("text-generation", model="
|
387 |
tts = TTS(model_name="tts_models/en/ljspeech/glow-tts", progress_bar=False, gpu=False)
|
388 |
|
389 |
# --------------- Feature 1: Daily Audio Digest ---------------
|
|
|
383 |
|
384 |
# --------------- Lightweight Models ---------------
|
385 |
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
386 |
+
story_gen = pipeline("text-generation", model="mrm8488/GPT-2-finetuned-fiction", max_length=250, pad_token_id=50256)
|
387 |
tts = TTS(model_name="tts_models/en/ljspeech/glow-tts", progress_bar=False, gpu=False)
|
388 |
|
389 |
# --------------- Feature 1: Daily Audio Digest ---------------
|