Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def load_model_and_tokenizer():
|
|
49 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
50 |
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=len(target_to_ind))
|
51 |
|
52 |
-
state_dict = safe_load("model.safetensors")
|
53 |
model.load_state_dict(state_dict)
|
54 |
|
55 |
return model, tokenizer
|
|
|
49 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
50 |
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=len(target_to_ind))
|
51 |
|
52 |
+
state_dict = safe_load("model (2).safetensors")
|
53 |
model.load_state_dict(state_dict)
|
54 |
|
55 |
return model, tokenizer
|