updates

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import open_clip
 import torch
 
-device = torch.device("cuda")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model, _, transform = open_clip.create_model_and_transforms(
     "coca_ViT-L-14",
-    pretrained="hf-hub:laion2B-s13B-b90k-mscoco-2014.pt"
+    pretrained="mscoco_finetuned_laion2B-s13B-b90k"
 )
 
 model.to(device)

requirements.txt

@@ -1,4 +1,4 @@
-git+https://github.com/mlfoundations/open_clip
+open_clip_torch
 transformers
 --extra-index-url https://download.pytorch.org/whl/cu113
 torch