Update app.py
Browse files
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)
|