Update app.py

app.py

@@ -94,10 +94,10 @@ class WaifuScorer:
         self.verbose = verbose
         if model_path is None:
             # Use default repo path – if the model file is not present locally, it will be downloaded.
-            model_path = "Eugeoter/waifu-scorer-v4-beta/model.pth"
+            model_path = "Eugeoter/waifu-scorer-v4-beta/model.safetensors"
         if not os.path.isfile(model_path):
             from huggingface_hub import hf_hub_download
-            model_path = hf_hub_download("Eugeoter/waifu-scorer-v4-beta", "model.pth", cache_dir=cache_dir)
+            model_path = hf_hub_download("Eugeoter/waifu-scorer-v4-beta", "model.safetensors", cache_dir=cache_dir)
         print(f"Loading pretrained WaifuScorer model from {model_path}")
         self.mlp = load_model(model_path, input_size=768, device=device)
         self.model2, self.preprocess = load_clip_models("ViT-L/14", device=device)