Spaces:
Runtime error
Runtime error
CorvaeOboro
commited on
Commit
β’
036cdc2
1
Parent(s):
66917e5
Update app.py
Browse files
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()
|