Spaces:

VOIDER
/

CIET

Runtime error

VOIDER commited on Feb 25

Commit

6c26275

verified ·

1 Parent(s): 77a6037

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ import inspect
 # Initialize the pipeline; if CUDA is available, use GPU (device=0), else CPU (device=-1)
 pipe_shadow = pipeline(
     "image-classification",
-    model="RE-N-Y/aesthetic-shadow-v2",
     device=0 if torch.cuda.is_available() else -1
 )

 # Initialize the pipeline; if CUDA is available, use GPU (device=0), else CPU (device=-1)
 pipe_shadow = pipeline(
     "image-classification",
+    model="NeoChen1024/aesthetic-shadow-v2-backup",
     device=0 if torch.cuda.is_available() else -1
 )