cpu
This commit is contained in:
parent
3624afb606
commit
1e82e1ddd0
2
demo.py
2
demo.py
|
@ -5,7 +5,7 @@ from torchvision.datasets import CIFAR100
|
||||||
import gradio as gr
|
import gradio as gr
|
||||||
|
|
||||||
# Load the model
|
# Load the model
|
||||||
device = "cuda" if torch.cuda.is_available() else "cpu"
|
device = "cpu"
|
||||||
model, preprocess = clip.load('ViT-B/32', device)
|
model, preprocess = clip.load('ViT-B/32', device)
|
||||||
|
|
||||||
# Download the dataset
|
# Download the dataset
|
||||||
|
|
Loading…
Reference in New Issue