Reyad-Ahmmed commited on
Commit
489e0e4
·
verified ·
1 Parent(s): dbd9d26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -190,6 +190,9 @@ else:
190
  #model = AutoModelForSequenceClassification.from_pretrained(model_name, subfolder="data-timeframe_model").to('cpu')
191
  #tokenizer = AutoTokenizer.from_pretrained(model_name, subfolder="data-timeframe_tokenizer")
192
 
 
 
 
193
  #Define the label mappings (this must match the mapping used during training)
194
  label_mapping = {
195
  0: "lastmonth",
 
190
  #model = AutoModelForSequenceClassification.from_pretrained(model_name, subfolder="data-timeframe_model").to('cpu')
191
  #tokenizer = AutoTokenizer.from_pretrained(model_name, subfolder="data-timeframe_tokenizer")
192
 
193
+ model = AutoModelForSequenceClassification.from_pretrained(model_save_path).to('cpu')
194
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_save_path)
195
+
196
  #Define the label mappings (this must match the mapping used during training)
197
  label_mapping = {
198
  0: "lastmonth",