Spaces:

isl-org
/

lang-seg

Runtime error

akhaliq HF staff commited on Apr 11, 2022

Commit

2f3e89a

1 Parent(s): f347ac1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -313,7 +313,7 @@ def load_model():
     model.std = [0.5, 0.5, 0.5]
     evaluator = LSeg_MultiEvalModule(
         model, scales=scales, flip=True
-    ).cuda()
     evaluator.eval()
     transform = transforms.Compose(

     model.std = [0.5, 0.5, 0.5]
     evaluator = LSeg_MultiEvalModule(
         model, scales=scales, flip=True
+    ).cpu()
     evaluator.eval()
     transform = transforms.Compose(