AkashKhamkar commited on
Commit
4398f74
·
1 Parent(s): afeccd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("CareerNinja/t5-large_3e-4")
24
  os.makedirs('./transcripts/')
25
 
26
  def clean_text(link,start,end):
@@ -182,7 +182,7 @@ 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 = []
 
20
 
21
 
22
  device = 'cuda' if cuda.is_available() else 'cpu'
23
+ tokenizer = AutoTokenizer.from_pretrained("CareerNinja/t5_large_3_1_3e_4_v3_dataset")
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_3_1_3e_4_v3_dataset")
186
  summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer)
187
  print(f""" Entered summarizer ! """)
188
  out = []