Spaces:
Paused
Paused
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",
|