Update app.py

app.py

@@ -8,8 +8,11 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 import sys
 import os
+import subprocess
 
 
+subprocess.Popen('git clone https://huggingface.co/SerdarHelli/BrainMRIGAN')
+
 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
     if grid_w is None:
@@ -30,7 +33,7 @@ def layout_grid(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True
 sys.path.insert(0, 'Brain-MR-Image-Generation-GAN ')
 
 
-network_pkl=hf_hub_download('SerdarHelli/BrainMRIGAN',f'models/braingan-400.pkl')
+network_pkl='braingan-400.pkl'
 with open(network_pkl, 'rb') as f:
     G = pickle.load(f)['G_ema']