Update app.py (#22)

- Update app.py (83aab9ce5f70ba14546ed34129cc33b6d022d0b0)

app.py

@@ -50,32 +50,33 @@ def upload_to_huggingface(repo_id, folder_path, oauth_token: Union[gr.OAuthToken
         return f"Upload failed: {str(e)}"
 
 def process_image_tag(image_tag, repo_id, oauth_token: Union[gr.OAuthToken, None]):
-    # Extract image and tag from the input
-    image, tag = image_tag.split(':')
-
-    # Fetch the manifest JSON
-    manifest_json = fetch_manifest(image, tag)
-    if not manifest_json or 'errors' in manifest_json:
-        return f"Failed to fetch the manifest for {image}:{tag}"
-
-    # Save the manifest JSON to the blobs folder
-    manifest_file_path = "blobs/manifest"
-    os.makedirs(os.path.dirname(manifest_file_path), exist_ok=True)
-    with open(manifest_file_path, 'w') as f:
-        json.dump(manifest_json, f)
-
-    # Extract the digest values from the JSON
-    digests = [layer['digest'] for layer in manifest_json.get('layers', [])]
-
-    # Download each file
-    for digest in digests:
-        download_file(digest, image)
-
-    # Download the config file
-    config_digest = manifest_json.get('config', {}).get('digest')
-    if config_digest:
-        download_file(config_digest, image)
     try:
+        # Extract image and tag from the input
+        image, tag = image_tag.split(':')
+    
+        # Fetch the manifest JSON
+        manifest_json = fetch_manifest(image, tag)
+        if not manifest_json or 'errors' in manifest_json:
+            return f"Failed to fetch the manifest for {image}:{tag}"
+    
+        # Save the manifest JSON to the blobs folder
+        manifest_file_path = "blobs/manifest"
+        os.makedirs(os.path.dirname(manifest_file_path), exist_ok=True)
+        with open(manifest_file_path, 'w') as f:
+            json.dump(manifest_json, f)
+    
+        # Extract the digest values from the JSON
+        digests = [layer['digest'] for layer in manifest_json.get('layers', [])]
+    
+        # Download each file
+        for digest in digests:
+            download_file(digest, image)
+    
+        # Download the config file
+        config_digest = manifest_json.get('config', {}).get('digest')
+        if config_digest:
+            download_file(config_digest, image)
+
         # Upload to Hugging Face Hub
         upload_result, repo_path = upload_to_huggingface(repo_id, 'blobs', oauth_token)