fix: no map location in main process

app.py

@@ -19,7 +19,7 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 
 use_gpu = True
 device = torch.device('cuda' if torch.cuda.is_available() and use_gpu else 'cpu')
-map_loc = None if torch.cuda.is_available() and use_gpu else 'cpu'
+# map_loc = None if torch.cuda.is_available() and use_gpu else 'cpu'
 
 # Inverse Cooking
 ingrs_vocab = pickle.load(
@@ -56,8 +56,11 @@ args.ingrs_only = False
         
 # Load the trained model parameters
 model = get_model(args, ingr_vocab_size, instrs_vocab_size)
+# model.load_state_dict(torch.load(
+#     hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN), map_location=map_loc)
+# )
 model.load_state_dict(torch.load(
-    hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN), map_location=map_loc)
+    hf_hub_download(REPO_ID, 'data/modelbest.ckpt', token=HF_TOKEN))
 )
 model.eval()
 model.ingrs_only = False