Update app.py

app.py

@@ -181,7 +181,7 @@ class Model_ML():
             model_LGB = joblib.load('model_LGB.pkl')
             model_XGB = joblib.load('model_XGB.pkl')
             model_RF = joblib.load('model_RF.pkl')
-            Net.load_state_dict(torch.load('model_ANN.pkl'))
+            Net.load_state_dict(torch.load('model_ANN.pkl')，map_location=torch.device('cpu'))
             return model_LGB,model_XGB,model_RF,Net
 
     def get_Rf(self):