Update app.py

app.py

@@ -1,12 +1,62 @@
-import gradio as gr
 import os
+import gradio as gr
+from huggingface_hub import hf_hub_download, login
+import torch
+
+try:
+    from safetensors import safe_open
+    SAFETENSORS_AVAILABLE = True
+except ImportError:
+    SAFETENSORS_AVAILABLE = False
+
+# Hugging Face Token（Secrets に登録しておく必要あり）
+HF_TOKEN = os.environ.get("HUGGINGFACE_HUB_TOKEN")
+if not HF_TOKEN:
+    raise RuntimeError("HUGGINGFACE_HUB_TOKEN が未設定です。SpacesのSecretsで設定してください。")
+
+login(token=HF_TOKEN)
+
+def check_model(repo_id, filename):
+    logs = []
+    try:
+        model_path = hf_hub_download(
+            repo_id=repo_id,
+            filename=filename,
+            token=HF_TOKEN
+        )
+        logs.append(f"📥 Downloaded: {model_path}")
+    except Exception as e:
+        return f"❌ Download failed: {str(e)}"
+
+    if SAFETENSORS_AVAILABLE and filename.endswith(".safetensors"):
+        try:
+            with safe_open(model_path, framework="pt", device="cpu") as f:
+                logs.append("✅ safetensors loaded successfully")
+                logs.append("Keys: " + ", ".join(list(f.keys())[:10]))
+                return "\n".join(logs)
+        except Exception as e:
+            logs.append(f"⚠ safetensors error: {e}")
+
+    try:
+        sd = torch.load(model_path, map_location="cpu")
+        logs.append("✅ torch.load OK")
+        if isinstance(sd, dict):
+            logs.append("Top-level keys: " + ", ".join(list(sd.keys())[:10]))
+        else:
+            logs.append(f"Loaded object type: {type(sd)}")
+    except Exception as e:
+        logs.append(f"❌ torch.load failed: {e}")
+    return "\n".join(logs)
 
-def check_token():
-    token = os.environ.get("HUGGINGFACE_HUB_TOKEN")
-    if token:
-        return "✅ トークン取得成功（最初の10文字）: " + token[:10] + "..."
-    else:
-        return "❌ トークンが取得できません。Spacesの Secrets に設定してください。"
+demo = gr.Interface(
+    fn=check_model,
+    inputs=[
+        gr.Textbox(label="Repo ID", value="revi13/ip-adapter-faceid-private"),
+        gr.Textbox(label="Filename", value="ip-adapter-faceid-plusv2_sd15.bin")
+    ],
+    outputs=gr.Text(label="Result"),
+    title="🧪 Hugging Face Hub モデル重みファイル検証ツール",
+    description="指定した repo_id / filename のモデル重みをダウンロードして torch.load/safetensors で検証します。"
+)
 
-demo = gr.Interface(fn=check_token, inputs=[], outputs="text")
 demo.launch()