Update app.py

app.py

@@ -14,6 +14,7 @@ import subprocess
 import os
 from typing import Tuple
 import PIL.Image
+from huggingface_hub import hf_hub_download
 
 os.system("git clone https://github.com/NVlabs/stylegan3")
 
@@ -26,10 +27,10 @@ DESCRIPTION = f'''This model generates healthy MR Brain Images.
 
 [Example]("https://huggingface.co/spaces/SerdarHelli/Brain-MR-Image-Generation-GAN/blob/main/ex.png")
 '''
+hf_hub_download("SerdarHelli/Brain-MRI-GAN", filename="brainmrigan.pkl")
 
 
-
-network_pkl='braingan-400.pkl'
+network_pkl='brainmrigan.pkl'
 with open(network_pkl, 'rb') as f:
     G = pickle.load(f)['G_ema'] 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")