Update app.py

app.py

@@ -68,7 +68,7 @@ class UNetWrapper:
         # Create a repository if it doesn't exist
         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
+        self.model.save_pretrained(self.repo_id,token=self.token)  # You may need to implement this method
 
 
 # Training function