download GCS model to to cache

app.py

@@ -57,13 +57,18 @@ def predict(img, confidence):
     # Download model from GCS
     model_path = os.getenv('MODEL_PATH')
     response = requests.get(model_path)
-    time.sleep(30)
+    time.sleep(20)
 
-    with open('model.pt', 'wb') as f:
+    # Define the cache directory path
+    cache_dir = os.path.expanduser("~/.cache/torch/hub/checkpoints")
+    model_file_name = os.path.basename(model_path)
+    model_file_path = os.path.join(cache_dir, model_file_name)
+    print(model_file_path)
+
+    with open(model_file_path, 'wb') as f:
         f.write(response.content)
-    time.sleep(30)
 
-    checkpoint = torch.load('model.pt', map_location=device)
+    checkpoint = torch.load(model_file_path, map_location=device)
     densenet.load_state_dict(checkpoint['model_state_dict'])
 
     densenet.eval()