Update main.py

main.py

@@ -13,7 +13,7 @@ def init():
     tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
     models = loading_model("bert-base-uncased")
     model = BirdModel_Attention_lstm(models, True)
-    checkpoint = torch.load("attention_lstm_pre.ckpt")
+    checkpoint = torch.load("attention_lstm_pre.ckpt",map_location=torch.device('cpu'))
     model.load_state_dict(checkpoint)
     return model, tokenizer