solving the device

app.py

@@ -15,7 +15,7 @@ def get_model_list():
 
 def load_model(path):
     model = JobFakeModel(base_model="distilbert", freeze_base=True)
-    model.load_state_dict(torch.load(path, device="cpu"))
+    model.load_state_dict(torch.load(path, map_location=torch.device('cpu')))
     return model
 
 st.title('Fake Jobs Streamlit App')