Reyad-Ahmmed commited on
Commit
04b6cfd
·
verified ·
1 Parent(s): f57b861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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')