Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -174,8 +174,11 @@ if (runModel=='1'):
|
|
| 174 |
tokenizer.save_pretrained('./' + modelNameToUse + '_tokenizer')
|
| 175 |
else:
|
| 176 |
print('Load Pre-trained')
|
| 177 |
-
model_save_path = "./" + modelNameToUse + "_model"
|
| 178 |
-
tokenizer_save_path = "./" + modelNameToUse + "_tokenizer"
|
|
|
|
|
|
|
|
|
|
| 179 |
|
| 180 |
# RobertaTokenizer.from_pretrained(model_save_path)
|
| 181 |
model = AutoModelForSequenceClassification.from_pretrained(model_save_path).to('cpu')
|
|
|
|
| 174 |
tokenizer.save_pretrained('./' + modelNameToUse + '_tokenizer')
|
| 175 |
else:
|
| 176 |
print('Load Pre-trained')
|
| 177 |
+
#model_save_path = "./" + modelNameToUse + "_model"
|
| 178 |
+
#tokenizer_save_path = "./" + modelNameToUse + "_tokenizer"
|
| 179 |
+
|
| 180 |
+
model_save_path = "Reyad-Ahmmed/hf-data-timeframe"
|
| 181 |
+
tokenizer_save_path = "Reyad-Ahmmed/hf-data-timeframe"
|
| 182 |
|
| 183 |
# RobertaTokenizer.from_pretrained(model_save_path)
|
| 184 |
model = AutoModelForSequenceClassification.from_pretrained(model_save_path).to('cpu')
|