Spaces:

VOIDER
/

CIET

Runtime error

VOIDER commited on Feb 25

Commit

24404d4

verified ·

1 Parent(s): 4193d6c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -151,7 +151,7 @@ class ImageEvaluationTool:
         # 1. Aesthetic Shadow
         print("Loading Aesthetic Shadow model...")
-        self.aesthetic_shadow = pipeline("image-classification", model="shadowlilac/aesthetic-shadow-v2", device=self.device)
         try:
             # 2. Waifu Scorer (requires CLIP)

         # 1. Aesthetic Shadow
         print("Loading Aesthetic Shadow model...")
+        self.aesthetic_shadow = pipeline("image-classification", model="NeoChen1024/aesthetic-shadow-v2-backup", device=self.device)
         try:
             # 2. Waifu Scorer (requires CLIP)