omkar90 commited on
Commit
61e95cf
·
verified ·
1 Parent(s): c040cff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ clip_device = "cuda" if torch.cuda.is_available() else "cpu"
26
  # os.makedirs("models", exist_ok=True)
27
  os.makedirs("/tmp/models", exist_ok=True)
28
 
29
- clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device, download_root="./models")
30
 
31
 
32
  # Class names
 
26
  # os.makedirs("models", exist_ok=True)
27
  os.makedirs("/tmp/models", exist_ok=True)
28
 
29
+ clip_model, clip_preprocess = clip.load("ViT-B/32", device=clip_device, download_root="/tmp/models")
30
 
31
 
32
  # Class names