Update app.py

app.py

@@ -6,6 +6,8 @@ import torch
 from tqdm import tqdm
 import gradio as gr 
 from huggingface_hub import hf_hub_download
+import sys
+
 
 def layout_grid(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True, to_numpy=True):
     batch_size, channels, img_h, img_w = img.shape
@@ -24,9 +26,12 @@ def layout_grid(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True
     return img
 
 
+sys.path.insert(0, 'stylegan_xl')
+
 
+TOKEN = os.environ['TOKEN']
 
-network_pkl=hf_hub_download('SerdarHelli/BrainMRIGAN','models/braingan-400.pkl')
+network_pkl=hf_hub_download('SerdarHelli/BrainMRIGAN',f'models/braingan-400.pkl',use_auth_token=TOKEN)
 with open(network_pkl, 'rb') as f:
     G = pickle.load(f)['G_ema']