Spaces:

swimmiing
/

ACL-SSL-zeroshot-demo

Running

swimmiing commited on Dec 13, 2023

Commit

8fee4bf

•

1 Parent(s): 00a76b6

Fix cpu only

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from PIL import Image
 def greet(image, audio):
-    device = "cuda:0" if torch.cuda.is_available() else "cpu"
     # Get model
     model_conf_file = f'./config/model/ACL_ViT16.yaml'

 def greet(image, audio):
+    device = "cpu"
     # Get model
     model_conf_file = f'./config/model/ACL_ViT16.yaml'