Update app.py

app.py

@@ -6,7 +6,9 @@ from torchvision import transforms
 from datasets import load_dataset
 from huggingface_hub import Repository
 from huggingface_hub import HfApi, HfFolder, Repository, create_repo
-
+import os
+token = os.getenv('HF_WRITE')
+print(token)
 import gradio as gr
 from PIL import Image
 import os
@@ -60,7 +62,7 @@ class UNetWrapper:
     def __init__(self, unet_model, repo_id):
         self.model = unet_model
         self.repo_id = repo_id
-        self.token = os.getenv('HF_WRITE')  # Make sure this environment variable is set
+        self.token = token  # Make sure this environment variable is set
         self.api = HfApi()
 
     def push_to_hub(self):
@@ -218,8 +220,8 @@ gr_interface = gr.Interface(
 )
 if __name__ == '__main__':
     # Create or clone the repository if necessary
-    repo = Repository(local_dir=model_repo_id, clone_from=model_repo_id)
-    repo.git_pull()
+    #repo = Repository(local_dir=model_repo_id, clone_from=model_repo_id)
+    #repo.git_pull()
 
     # Launch the Gradio app
     gr_interface.launch()