Reyad-Ahmmed commited on
Commit
92f652a
·
verified ·
1 Parent(s): 978df6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -177,8 +177,14 @@ else:
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')
 
177
  #model_save_path = "./" + modelNameToUse + "_model"
178
  #tokenizer_save_path = "./" + modelNameToUse + "_tokenizer"
179
 
180
+ model_name = "Reyad-Ahmmed/hf-data-timeframe"
181
+
182
+ # Load the tokenizer from the subdirectory
183
+ tokenizer_save_path = AutoTokenizer.from_pretrained(model_name, subfolder="data-timeframe_tokenizer")
184
+
185
+ # Load the model from the subdirectory
186
+ model_save_path = AutoModel.from_pretrained(model_name, subfolder="data-timeframe_model")
187
+
188
 
189
  # RobertaTokenizer.from_pretrained(model_save_path)
190
  model = AutoModelForSequenceClassification.from_pretrained(model_save_path).to('cpu')