Abijith commited on
Commit
ecff4a2
1 Parent(s): 55d9e6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,9 +6,10 @@ import os
6
  import gradio as gr
7
  from transformers import AutoTokenizer, T5ForConditionalGeneration
8
 
 
9
  # Load model directly
10
- tokenizer = AutoTokenizer.from_pretrained("Abijith/my_first_t5_billsum_model")
11
- model = T5ForConditionalGeneration.from_pretrained("Abijith/my_first_t5_billsum_model")
12
 
13
  def summarize_text(input_text):
14
  summar_input = 'summarize: '+input_text
 
6
  import gradio as gr
7
  from transformers import AutoTokenizer, T5ForConditionalGeneration
8
 
9
+ model_name = 'Abijith/Billsum-text-summarizer-t5-small'
10
  # Load model directly
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
12
+ model = T5ForConditionalGeneration.from_pretrained(model_name)
13
 
14
  def summarize_text(input_text):
15
  summar_input = 'summarize: '+input_text