Update app.py

app.py

@@ -10,7 +10,10 @@ ind_to_target = {ind: target for target, ind in target_to_ind.items()}
 def load_model_and_tokenizer():
     model_name = 'distilbert/distilbert-base-cased'
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSequenceClassification.from_pretrained("./model.safetensors", num_labels=len(target_to_ind))
+    model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=len(target_to_ind))
+    
+    state_dict = safe_load("model.safetensors")
+    model.load_state_dict(state_dict)
     
     return model, tokenizer