AkashKhamkar commited on
Commit
e0d02fc
1 Parent(s): fb101c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -202,8 +202,8 @@ def clean_text(link,start,end):
202
  def t5_summarizer(link,start, end):
203
  input_text = clean_text(link,start,end)
204
  lst_outputs = []
205
- tokenizer1 = AutoTokenizer.from_pretrained("CareerNinja/t5-large_3e-4")
206
- model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5-large_3e-4")
207
  summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer1)
208
  print(f""" Entered summarizer ! """)
209
  st.write('Below is the summary of the given URL: ')
 
202
  def t5_summarizer(link,start, end):
203
  input_text = clean_text(link,start,end)
204
  lst_outputs = []
205
+ tokenizer1 = AutoTokenizer.from_pretrained("CareerNinja/t5_large_3e-4_on_v2_dataset")
206
+ model1 = AutoModelForSeq2SeqLM.from_pretrained("CareerNinja/t5_large_3e-4_on_v2_dataset")
207
  summarizer1 = pipeline("summarization", model=model1, tokenizer=tokenizer1)
208
  print(f""" Entered summarizer ! """)
209
  st.write('Below is the summary of the given URL: ')