cpu

2021-03-23 13:50:30 -04:00 · 2021-03-23 13:50:30 -04:00 · 1e82e1ddd0
parent 3624afb606
commit 1e82e1ddd0
1 changed files with 1 additions and 1 deletions
--- a/demo.py
+++ b/demo.py
@ -5,7 +5,7 @@ from torchvision.datasets import CIFAR100
 import gradio as gr

 # Load the model
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cpu"
 model, preprocess = clip.load('ViT-B/32', device)

 # Download the dataset