Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ class UNetWrapper:
|
|
64 |
self.model = unet_model
|
65 |
self.repo_id = repo_id
|
66 |
self.token = os.getenv('NEW_TOKEN') # Make sure this environment variable is set
|
67 |
-
self.api = HfApi()
|
68 |
|
69 |
def push_to_hub(self):
|
70 |
try:
|
|
|
64 |
self.model = unet_model
|
65 |
self.repo_id = repo_id
|
66 |
self.token = os.getenv('NEW_TOKEN') # Make sure this environment variable is set
|
67 |
+
self.api = HfApi(token=os.getenv('NEW_TOKEN'))
|
68 |
|
69 |
def push_to_hub(self):
|
70 |
try:
|