VOIDER commited on
Commit
24404d4
·
verified ·
1 Parent(s): 4193d6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="shadowlilac/aesthetic-shadow-v2", device=self.device)
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)