Update app.py
Browse files
app.py
CHANGED
@@ -151,7 +151,7 @@ class ImageEvaluationTool:
|
|
151 |
|
152 |
# 1. Aesthetic Shadow
|
153 |
print("Loading Aesthetic Shadow model...")
|
154 |
-
self.aesthetic_shadow = pipeline("image-classification", model="
|
155 |
|
156 |
try:
|
157 |
# 2. Waifu Scorer (requires CLIP)
|
|
|
151 |
|
152 |
# 1. Aesthetic Shadow
|
153 |
print("Loading Aesthetic Shadow model...")
|
154 |
+
self.aesthetic_shadow = pipeline("image-classification", model="NeoChen1024/aesthetic-shadow-v2-backup", device=self.device)
|
155 |
|
156 |
try:
|
157 |
# 2. Waifu Scorer (requires CLIP)
|