Spaces:
Running
Running
yapf
Browse files
model.py
CHANGED
@@ -15,7 +15,6 @@ HF_TOKEN = os.environ['HF_TOKEN']
|
|
15 |
|
16 |
|
17 |
class Model:
|
18 |
-
|
19 |
def __init__(self, device: str | torch.device):
|
20 |
self.device = torch.device(device)
|
21 |
self.model = self.load_model('stylegan_human_v2_1024.pkl')
|
|
|
15 |
|
16 |
|
17 |
class Model:
|
|
|
18 |
def __init__(self, device: str | torch.device):
|
19 |
self.device = torch.device(device)
|
20 |
self.model = self.load_model('stylegan_human_v2_1024.pkl')
|