Update app.py

app.py

@@ -2,8 +2,8 @@ from transformers import DistilBertForQuestionAnswering, DistilBertConfig, Disti
 import torch
 
 model = DistilBertForQuestionAnswering(DistilBertConfig.from_pretrained('distilbert/distilbert-base-multilingual-cased')).to("cpu")
-st_dict = torch.load("QazDistilBERT.pt")
-model.load_state_dict(st_dict,map_location=torch.device('cpu'))
+st_dict = torch.load("QazDistilBERT.pt",map_location=torch.device('cpu'))
+model.load_state_dict(st_dict)
 tokenizer = DistilBertTokenizerFast.from_pretrained("dappyx/QazDistilbertFast-tokenizerV3")
 
 import gradio as gr