Update app.py

app.py

@@ -11,7 +11,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 weight_dtype = torch.bfloat16 if device == "cuda" else torch.float32
 print(f"Using device: {device} ({weight_dtype})")
 print("Loading model...")
-model_pretrained_name_or_path = "kakaobrain/align-base"
+model_pretrained_name_or_path = "laion/CLIP-convnext_large_d.laion2B-s26B-b102K-augreg"
 processor = AutoProcessor.from_pretrained(model_pretrained_name_or_path)
 model = AutoModel.from_pretrained(model_pretrained_name_or_path, torch_dtype=weight_dtype).eval().to(device)
 print("Model loaded.")