Bhaskar2611 commited on
Commit
1379ec0
·
verified ·
1 Parent(s): 39ea8b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="distilgpt2", 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 ---------------
 
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 ---------------