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')
|