Hugging Face's logo

Spaces:
SerdarHelli
/
Brain-MR-Image-Generation-with-StyleGAN
Paused

App Files Community
3
SerdarHelli commited on
Commit
334e27d
·
1 Parent(s): 83b0577

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -27,10 +27,9 @@ DESCRIPTION = f'''This model generates healthy MR Brain Images.
27
 
28
  [Example]("https://huggingface.co/spaces/SerdarHelli/Brain-MR-Image-Generation-GAN/blob/main/ex.png")
29
  '''
30
- hf_hub_download("SerdarHelli/Brain-MRI-GAN", filename="brainmrigan.pkl",cache_dir="./model",revision="main")
31
 
32
 
33
- network_pkl='./model/61b352960b95f66c2b1dd346ea9ab1ccabfcfc23b51977cbfd301889bf7304ef.abd9063484d155f44754cefcb942e98bba5da8aa506173e68fb8f1439acdb419.pkl'
34
  with open(network_pkl, 'rb') as f:
35
  G = pickle.load(f)['G_ema']
36
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
27
 
28
  [Example]("https://huggingface.co/spaces/SerdarHelli/Brain-MR-Image-Generation-GAN/blob/main/ex.png")
29
  '''
30
+ network_pkl=hf_hub_download("SerdarHelli/Brain-MRI-GAN", filename="brainmrigan.pkl",revision="main")
31
 
32
 
 
33
  with open(network_pkl, 'rb') as f:
34
  G = pickle.load(f)['G_ema']
35
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")