Update app.py

app.py

@@ -1,12 +1,18 @@
+from huggingface_hub import snapshot_download
 import os
 import json
 import shutil
 import gradio as gr
-from huggingface_hub import snapshot_download
 
 def process_and_zip_folders(huggingface_dataset_url, output_dir):
+    # URLからリポジトリIDを抽出
+    if huggingface_dataset_url.startswith("https://huggingface.co/"):
+        repo_id = huggingface_dataset_url.replace("https://huggingface.co/", "").rstrip("/")
+    else:
+        repo_id = huggingface_dataset_url
+
     # データセットをダウンロード
-    dataset_path = snapshot_download(huggingface_dataset_url)
+    dataset_path = snapshot_download(repo_id)
 
     # 処理対象のフォルダ
     folders = [