Update gradio_demo/app.py

gradio_demo/app.py

@@ -36,7 +36,13 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
         input_image = Image.fromarray(res)
     return input_image
 
-instantir_path = os.environ['INSTANTIR_PATH']
+from huggingface_hub import hf_hub_download
+
+hf_hub_download(repo_id="InstantX/InstantIR", filename="models/adapter.pt", local_dir=".")
+hf_hub_download(repo_id="InstantX/InstantIR", filename="models/aggregator.pt", local_dir=".")
+hf_hub_download(repo_id="InstantX/InstantIR", filename="models/previewer_lora_weights.bin", local_dir=".")
+
+instantir_path = f'./models'
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 sdxl_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"