ZinebSN commited on
Commit
a7bbdab
·
1 Parent(s): a224e26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,8 +14,9 @@ selected_model = st.radio("Select Model", list(models.keys()))
14
  model_name = models[selected_model]
15
 
16
  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)
 
14
  model_name = models[selected_model]
15
 
16
  if model_name == "T5 Small":
 
17
  tokenizer = AutoTokenizer.from_pretrained('t5-small')
18
+ model = T5().from_pretrained(model_name)
19
+
20
  #else:
21
  #model = GPT2().from_pretrained(model_name)
22
  #tokenizer = AutoTokenizer.from_pretrained(model_name)