ZinebSN commited on
Commit
bcff816
·
1 Parent(s): 9eadd05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ if model_name == "T5 Small":
17
  model = T5().from_pretrained(model_name)
18
  tokenizer = AutoTokenizer.from_pretrained('t5-small')
19
  else:
20
- model = GPT2().from_pretrained(model_name)
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22
 
23
  input_text=st.text_area("Input the text to summarize","")
 
17
  model = T5().from_pretrained(model_name)
18
  tokenizer = AutoTokenizer.from_pretrained('t5-small')
19
  else:
20
+ #model = GPT2().from_pretrained(model_name)
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22
 
23
  input_text=st.text_area("Input the text to summarize","")