Sristi222 commited on
Commit
4f01514
·
verified ·
1 Parent(s): 804ffb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -18,7 +18,9 @@ except ModuleNotFoundError:
18
  from youtube_transcript_api._errors import VideoUnavailable, TranscriptsDisabled
19
 
20
  # Model path and pipeline setup
21
- #model_path = "../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
 
 
22
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
23
 
24
  # Function to summarize text
 
18
  from youtube_transcript_api._errors import VideoUnavailable, TranscriptsDisabled
19
 
20
  # Model path and pipeline setup
21
+ # model_path = "../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
22
+ # text_summary = pipeline("summarization", model=model_path, torch_dtype=torch.bfloat16)
23
+
24
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
25
 
26
  # Function to summarize text