Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|