Spaces:

Shio-Koube
/

Demo-Anime

Sleeping

Shio-Koube commited on Feb 13

Commit

5f2b8e8

verified ·

1 Parent(s): c0365b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ from io import BytesIO
 MODEL_ARCH = "animetimm/convnext_base.dbv4-full"
 CLASSES = ["Good", "Normal", "Bad"]  # Must match your training order
 MODEL_PATH = "best_model.pth"
-MODEL_URL = "https://huggingface.co/Shio-Koube/Anime-rater-2/resolve/main/best_model.pth"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 # Cap the longest side to avoid wasting GPU compute

 MODEL_ARCH = "animetimm/convnext_base.dbv4-full"
 CLASSES = ["Good", "Normal", "Bad"]  # Must match your training order
 MODEL_PATH = "best_model.pth"
+MODEL_URL = "https://huggingface.co/Shio-Koube/ConvNext-aesthetic-rater/resolve/main/best_model.pth"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 # Cap the longest side to avoid wasting GPU compute