Update app.py (#11)

- Update app.py (931f37e9ab5b1e2744612609c6c22aabfde7f7ba)

app.py

@@ -30,7 +30,8 @@ def fetch_manifest(image, tag):
     else:
         return None
 
-def upload_to_huggingface(repo_id, folder_path, token):
+def upload_to_huggingface(repo_id, folder_path, oauth_token: Union[gr.OAuthToken, None])):
+    token = oauth_token.token if oauth_token else None
     api = HfApi(token=token)
     repo_path = api.create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)
     print(f"Repo created {repo_path}")
@@ -47,8 +48,6 @@ def upload_to_huggingface(repo_id, folder_path, token):
 def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None]):
     # Extract image and tag from the input
     try:
-        token = oauth_token.token if oauth_token else None
-        
         image, tag = image_tag.split(':')
     
         # Fetch the manifest JSON
@@ -75,7 +74,7 @@ def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None
             download_file(config_digest, image)
     
         # Upload to Hugging Face Hub
-        upload_result = upload_to_huggingface(repo_id, 'blobs/*', token=token)
+        upload_result = upload_to_huggingface(repo_id, 'blobs/*', oauth_token)
     
         # Delete the blobs folder
         shutil.rmtree('blobs')