Update app.py

app.py

@@ -60,12 +60,13 @@ class UNetWrapper:
     def __init__(self, unet_model, repo_id):
         self.model = unet_model
         self.repo_id = repo_id
+        self.token = os.getenv('HF_WRITE')
 
     def push_to_hub(self):
         # Initialize the Hugging Face API
         api = HfApi()
         # Create a repository if it doesn't exist
-        create_repo(self.repo_id, exist_ok=True)
+        create_repo(self.repo_id, exist_ok=True,token=self.token)
         # Push the model's state dict to the Hugging Face Hub
         self.model.save_pretrained(self.repo_id)  # You may need to implement this method