From 1e82e1ddd07fb86b8b76263b0ad67e7605c49bb3 Mon Sep 17 00:00:00 2001
From: AK391 <ahsengradio@gmail.com>
Date: Tue, 23 Mar 2021 13:50:30 -0400
Subject: [PATCH] cpu

---
 demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo.py b/demo.py
index 772fc91..fce2835 100644
--- 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