Update app.py

app.py

@@ -5,12 +5,23 @@ from torchvision import transforms
 from PIL import Image
 
 # モデル読み込み
+from huggingface_hub import hf_hub_download
+import torch
+import timm
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+model_path = hf_hub_download(
+    repo_id="ryumiyake/ips-model-weights",  # ←作ったModel Hubのリポジトリ名
+    filename="model.pth"
+)
+
 model = timm.create_model('convnext_small', pretrained=False, num_classes=2)
-model.load_state_dict(torch.load("model.pth", map_location=device))
+model.load_state_dict(torch.load(model_path, map_location=device))
 model.to(device)
 model.eval()
 
+
 # 前処理
 transform = transforms.Compose([
     transforms.Resize((224, 224)),