[feat] Check for model entry

app.py

@@ -4,6 +4,7 @@ import os
 import subprocess
 from PIL import Image
 from huggingface_hub import hf_hub_download
+from shutil import copyfile
 
 UPLOAD_DIR = "./uploaded_images"
 RESULTS_DIR = "./results"
@@ -18,6 +19,10 @@ MODEL_FILE = "310_net_G.pth"
 
 model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILE, cache_dir=CHECKPOINTS_DIR)
 
+expected_model_path = os.path.join(CHECKPOINTS_DIR, MODEL_FILE)
+if not os.path.exists(expected_model_path):
+    copyfile(model_path, expected_model_path)
+
 def reflection_removal(input_images):
     for input_image in input_images:
         file_path = os.path.join(UPLOAD_DIR, input_image.name)