Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ from tqdm import tqdm
|
|
| 27 |
# Config - update these IDs as needed
|
| 28 |
# -------------------------
|
| 29 |
SIGLIP_MODEL_ID = "EYEDOL/siglipFULL-agri-finetuned"
|
| 30 |
-
LLAVA_MODEL_ID = "
|
| 31 |
DATASET_TEMPLATE = "EYEDOL/AGRILLAVA-image-text{}"
|
| 32 |
NUM_DATASETS = 1 # set to 15 if you want all datasets (startup memory/time increases)
|
| 33 |
BATCH_SIZE = 16
|
|
|
|
| 27 |
# Config - update these IDs as needed
|
| 28 |
# -------------------------
|
| 29 |
SIGLIP_MODEL_ID = "EYEDOL/siglipFULL-agri-finetuned"
|
| 30 |
+
LLAVA_MODEL_ID = "liuhaotian/llava-v1.6-vicuna-7b"
|
| 31 |
DATASET_TEMPLATE = "EYEDOL/AGRILLAVA-image-text{}"
|
| 32 |
NUM_DATASETS = 1 # set to 15 if you want all datasets (startup memory/time increases)
|
| 33 |
BATCH_SIZE = 16
|