fix load model

app.py

@@ -30,7 +30,7 @@ id_to_description = load_tags_info()
 def load_model():
     config = DistilBertConfig.from_json_file('./config.json')
     model = DistilBertForSequenceClassification(config)
-    state_dict = torch.load('./pytorch_model.bin')
+    state_dict = torch.load('./pytorch_model.bin', map_location=torch.device('cpu'))
     model.load_state_dict(state_dict)
     return model