Spaces:

ZinebSN
/

T5_Small1

Runtime error

ZinebSN commited on May 28, 2023

Commit

75bbfe8

1 Parent(s): a7bbdab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,9 +12,9 @@ models = {
 selected_model = st.radio("Select Model", list(models.keys()))
 model_name = models[selected_model]
 if model_name == "T5 Small":
-    tokenizer = AutoTokenizer.from_pretrained('t5-small')
     model = T5().from_pretrained(model_name)
 #else:

 selected_model = st.radio("Select Model", list(models.keys()))
 model_name = models[selected_model]
+tokenizer = AutoTokenizer.from_pretrained('t5-small')
 if model_name == "T5 Small":
     model = T5().from_pretrained(model_name)
 #else: