K00B404 commited on
Commit
a82ec09
·
verified ·
1 Parent(s): 275c6f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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: