Spaces:
Sleeping
Sleeping
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')
|