Modify device parameter

app.py

@@ -114,7 +114,7 @@ def load_kmeans_model():
 @st.cache_resource(show_spinner="Loading SimilarityCal model...")
 def load_similaritycal_model():
     sim_cal_model = PairClassifier()
-    sim_cal_model.load_state_dict(torch.load(SIMILARITY_CAL_MODEL_PATH))
+    sim_cal_model.load_state_dict(torch.load(SIMILARITY_CAL_MODEL_PATH, map_location=device))
     sim_cal_model = sim_cal_model.to(device)
     sim_cal_model = sim_cal_model.eval()
     return sim_cal_model