CorvaeOboro commited on
Commit
036cdc2
β€’
1 Parent(s): 66917e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -55,9 +55,7 @@ def generate_image(seed: int, truncation_psi: float, model: nn.Module,
55
 
56
 
57
  def load_model(file_name: str, device: torch.device) -> nn.Module:
58
- path = hf_hub_download(f'CorvaeOboro/gen_ability_icon',
59
- f'{file_name}',
60
- use_auth_token=TOKEN)
61
  with open(path, 'rb') as f:
62
  model = pickle.load(f)['G_ema']
63
  model.eval()
 
55
 
56
 
57
  def load_model(file_name: str, device: torch.device) -> nn.Module:
58
+ path = hf_hub_download(f'CorvaeOboro/gen_ability_icon' , f'{file_name}')
 
 
59
  with open(path, 'rb') as f:
60
  model = pickle.load(f)['G_ema']
61
  model.eval()