AkashKhamkar commited on
Commit
afeccd2
·
1 Parent(s): 7a3e1ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -20,7 +20,7 @@ import gradio as gr
20
 
21
 
22
  device = 'cuda' if cuda.is_available() else 'cpu'
23
- tokenizer = AutoTokenizer.from_pretrained("t5-base")
24
  os.makedirs('./transcripts/')
25
 
26
  def clean_text(link,start,end):
@@ -182,9 +182,8 @@ def clean_text(link,start,end):
182
 
183
  def t5_summarizer(link,start, end):
184
  input_text = clean_text(link,start,end)
185
- tokenizer1 = AutoTokenizer.from_pretrained("CareerNinja/t5-large_3e-4")
186
  model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5-large_3e-4")
187
- summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer1)
188
  print(f""" Entered summarizer ! """)
189
  out = []
190
  for i in range(len(input_text)):
 
20
 
21
 
22
  device = 'cuda' if cuda.is_available() else 'cpu'
23
+ tokenizer = AutoTokenizer.from_pretrained("CareerNinja/t5-large_3e-4")
24
  os.makedirs('./transcripts/')
25
 
26
  def clean_text(link,start,end):
 
182
 
183
  def t5_summarizer(link,start, end):
184
  input_text = clean_text(link,start,end)
 
185
  model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5-large_3e-4")
186
+ summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer)
187
  print(f""" Entered summarizer ! """)
188
  out = []
189
  for i in range(len(input_text)):