Update app.py
Browse files
app.py
CHANGED
|
@@ -47,13 +47,13 @@ def load_models():
|
|
| 47 |
|
| 48 |
|
| 49 |
# ---------- AGRICLIP ----------
|
|
|
|
| 50 |
agri_model, _, agri_preprocess = open_clip.create_model_and_transforms(
|
| 51 |
-
|
| 52 |
-
|
|
|
|
| 53 |
|
| 54 |
-
agri_tokenizer = open_clip.get_tokenizer(
|
| 55 |
-
"hf-hub:microsoft/AgriCLIP"
|
| 56 |
-
)
|
| 57 |
|
| 58 |
agri_model = agri_model.to(device).eval()
|
| 59 |
|
|
|
|
| 47 |
|
| 48 |
|
| 49 |
# ---------- AGRICLIP ----------
|
| 50 |
+
# ---------- AGRICULTURE CLIP ----------
|
| 51 |
agri_model, _, agri_preprocess = open_clip.create_model_and_transforms(
|
| 52 |
+
"ViT-B-32",
|
| 53 |
+
pretrained="laion2b_s34b_b79k"
|
| 54 |
+
)
|
| 55 |
|
| 56 |
+
agri_tokenizer = open_clip.get_tokenizer("ViT-B-32")
|
|
|
|
|
|
|
| 57 |
|
| 58 |
agri_model = agri_model.to(device).eval()
|
| 59 |
|