Update app.py

app.py

@@ -27,10 +27,9 @@ 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="./model",revision="main")
+network_pkl=hf_hub_download("SerdarHelli/Brain-MRI-GAN", filename="brainmrigan.pkl",revision="main")
 
 
-network_pkl='./model/61b352960b95f66c2b1dd346ea9ab1ccabfcfc23b51977cbfd301889bf7304ef.abd9063484d155f44754cefcb942e98bba5da8aa506173e68fb8f1439acdb419.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")