Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -464,8 +464,7 @@ from TTS.api import TTS
|
|
464 |
|
465 |
# --------------- Lightweight Models ---------------
|
466 |
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
467 |
-
|
468 |
-
story_gen = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", max_length=250, pad_token_id=50256)
|
469 |
set_seed(42)
|
470 |
tts = TTS(model_name="tts_models/en/ljspeech/glow-tts", progress_bar=False, gpu=False)
|
471 |
|
|
|
464 |
|
465 |
# --------------- Lightweight Models ---------------
|
466 |
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
467 |
+
story_gen = pipeline("text-generation", model="aspis/gpt2-genre-story-generation", max_length=250, pad_token_id=50256)
|
|
|
468 |
set_seed(42)
|
469 |
tts = TTS(model_name="tts_models/en/ljspeech/glow-tts", progress_bar=False, gpu=False)
|
470 |
|