Spaces:

MR
/

vilt-word-patch-alignment

Build error

MR commited on Oct 5, 2022

Commit

a46cb3f

•

1 Parent(s): 5e18de6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ def main(_config):
     model.setup("test")
     model.eval()
-    device = "cuda:0" if _config["num_gpus"] > 0 else "cpu"
     model.to(device)
     def infer(url, mp_text, hidx):

     model.setup("test")
     model.eval()
+    device = "cpu"
     model.to(device)
     def infer(url, mp_text, hidx):