Walid Ahmed commited on
Commit
29cae24
1 Parent(s): b73d62b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -7,10 +7,7 @@ from transformers import pipeline
7
 
8
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
 
10
- # model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots"
11
- # "/a4f8f3ea906ed274767e9906dbaede7531d660ff")
12
- # text_summary = pipeline("summarization", model=model_path,
13
- # torch_dtype=torch.bfloat16)
14
 
15
  def summary (input):
16
  output = text_summary(input)
@@ -44,9 +41,6 @@ def get_youtube_transcript(video_url):
44
  return f"An error occurred: {e}"
45
 
46
 
47
- # Example URL (Replace this with the actual URL when using the script)
48
- # video_url = "https://youtu.be/5PibknhIsTc"
49
- # print(get_youtube_transcript(video_url))
50
 
51
  gr.close_all()
52
 
 
7
 
8
  text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
 
10
+ torch_dtype=torch.bfloat16)
 
 
 
11
 
12
  def summary (input):
13
  output = text_summary(input)
 
41
  return f"An error occurred: {e}"
42
 
43
 
 
 
 
44
 
45
  gr.close_all()
46