Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,7 @@ model_names = ['distilbert-base-uncased-finetuned-sst-2-english', 'bert-base-unc
|
|
| 23 |
model_name = st.selectbox("Select a pretrained model", model_names)
|
| 24 |
|
| 25 |
if model_name == "Codys-Finetuning-Language-Model":
|
| 26 |
-
model = transformers.BertForSequenceClassification.from_pretrained("./
|
| 27 |
tokenizer = transformers.BertTokenizerFast.from_pretrained("bert-base-uncased")
|
| 28 |
else:
|
| 29 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 23 |
model_name = st.selectbox("Select a pretrained model", model_names)
|
| 24 |
|
| 25 |
if model_name == "Codys-Finetuning-Language-Model":
|
| 26 |
+
model = transformers.BertForSequenceClassification.from_pretrained("./myModel/")
|
| 27 |
tokenizer = transformers.BertTokenizerFast.from_pretrained("bert-base-uncased")
|
| 28 |
else:
|
| 29 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
|