fix version

app.py

@@ -1,25 +1,29 @@
+
 import spaces
 import os
+import gradio as gr
 import json
 import torch
-import requests
-import subprocess
-import gradio as gr
-from omegaconf import OmegaConf
+
+from huggingface_hub import snapshot_download
+
+from inference_bokehK import load_models as load_bokeh_models, run_inference as run_bokeh_inference, OmegaConf
+from inference_focal_length import load_models as load_focal_models, run_inference as run_focal_inference
+from inference_shutter_speed import load_models as load_shutter_models, run_inference as run_shutter_inference
+from inference_color_temperature import load_models as load_color_models, run_inference as run_color_inference
+
+
 
 
 model_path = "ckpts"
 os.makedirs(model_path, exist_ok=True)
 
-# Download files from Hugging Face repository using wget
-def download_huggingface_files():
-    # URL to the Hugging Face repository
-    repo_url = "https://huggingface.co/pandaphd/generative_photography/tree/main/"
-    # Using wget to download all the files from the given Hugging Face repository
-    subprocess.run(["wget", "-r", "-np", "-nH", "--cut-dirs=3", "-P", model_path, repo_url])
 
 print("Downloading models from Hugging Face...")
-download_huggingface_files()
+snapshot_download(repo_id="pandaphd/generative_photography", local_dir=model_path)
+
+
+
 
 
 torch.manual_seed(42)

app_old.py → app_wget.py

@@ -1,29 +1,25 @@
-
 import spaces
 import os
-import gradio as gr
 import json
 import torch
-
-from huggingface_hub import snapshot_download
-
-from inference_bokehK import load_models as load_bokeh_models, run_inference as run_bokeh_inference, OmegaConf
-from inference_focal_length import load_models as load_focal_models, run_inference as run_focal_inference
-from inference_shutter_speed import load_models as load_shutter_models, run_inference as run_shutter_inference
-from inference_color_temperature import load_models as load_color_models, run_inference as run_color_inference
-
-
+import requests
+import subprocess
+import gradio as gr
+from omegaconf import OmegaConf
 
 
 model_path = "ckpts"
 os.makedirs(model_path, exist_ok=True)
 
+# Download files from Hugging Face repository using wget
+def download_huggingface_files():
+    # URL to the Hugging Face repository
+    repo_url = "https://huggingface.co/pandaphd/generative_photography/tree/main/"
+    # Using wget to download all the files from the given Hugging Face repository
+    subprocess.run(["wget", "-r", "-np", "-nH", "--cut-dirs=3", "-P", model_path, repo_url])
 
 print("Downloading models from Hugging Face...")
-snapshot_download(repo_id="pandaphd/generative_photography", local_dir=model_path)
-
-
-
+download_huggingface_files()
 
 
 torch.manual_seed(42)

requirements.txt

@@ -7,6 +7,7 @@ imageio-ffmpeg
 numpy==1.24.4
 transformers==4.39.3
 accelerate==0.30.0
+huggingface_hub==0.30.0
 opencv-python
 gdown
 einops