Update app.py

app.py

@@ -35,7 +35,8 @@ model_name = "trained_model2"
 model_path = model_name + '.zip'
 model.load_state_dict(
     torch.load(
-        model_path
+        model_path,
+        map_location=torch.device("cpu")
     )
 )