Update app.py

app.py

@@ -27,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",cache_dir="")
+hf_hub_download("SerdarHelli/Brain-MRI-GAN", filename="brainmrigan.pkl",cache_dir="./model")
 
 
-network_pkl='brainmrigan.pkl'
+network_pkl='./model/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")