dfedukov commited on
Commit
1fbf315
·
verified ·
1 Parent(s): 683615c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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