change small model

app.py

@@ -89,8 +89,8 @@ def zeroshot_classifier(model, classnames, templates, device):
         zeroshot_weights = torch.stack(zeroshot_weights).cuda()
     return zeroshot_weights*100
 
-# device = "cuda" if torch.cuda.is_available() else "cpu"
-device = "cuda"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+# device = "cuda"
 # Instantiate model
 vis_model = CLIPModel_Super("ViT-B/16", device=device, download_root="./ckpt")
 vis_model.eval()