Bhaskar2611 commited on
Commit
b4a4fe2
·
verified ·
1 Parent(s): d9997f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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
- # Use GPT-Neo 125M for narrative generation
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