localhorst
/
CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cpu

This commit is contained in:
AK391 2021-03-23 13:50:30 -04:00
parent 3624afb606
commit 1e82e1ddd0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo.py
View File

@ -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