Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 import gradio as gr
 from PIL import Image
+from huggingface_hub import hf_hub_url, cached_download
+
 
 os.system("git clone https://github.com/AK391/stylegan2-ada-pytorch")
 
@@ -10,10 +12,11 @@ os.chdir("stylegan2-ada-pytorch")
 os.mkdir("outputs")
 os.mkdir("outputs/images")
 
-os.system("gdown --id '1DqbBf6298ech_o9Z-N2IA0FDxtw5_rlk'")
+config_file_url = hf_hub_url("AUBMC-AIM/OCTaGAN", filename="OCTaGAN.pkl")
+cached_file = cached_download(config_file_url)
 
 def inference(truncation,seeds):
-  os.system("python generate.py --outdir=./outputs/images/ --trunc="+str(truncation)+" --seeds="+str(int(seeds))+" --network=OCTaGAN.pkl")
+  os.system("python generate.py --outdir=./outputs/images/ --trunc="+str(truncation)+" --seeds="+str(int(seeds))+" --network="+cached_file)
   seeds = int(seeds)
   image = Image.open(f"./outputs/images/seed{seeds:04d}.png")
   return image