Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ from transformers import AutoProcessor, AutoModel
|
|
| 25 |
from tqdm import tqdm
|
| 26 |
|
| 27 |
SIGLIP_MODEL_ID = "EYEDOL/siglipFULL-agri-finetuned"
|
| 28 |
-
LLAVA_MODEL_ID = "
|
| 29 |
DATASET_TEMPLATE = "EYEDOL/AGRILLAVA-image-text{}"
|
| 30 |
NUM_DATASETS = 1
|
| 31 |
BATCH_SIZE = 16
|
|
|
|
| 25 |
from tqdm import tqdm
|
| 26 |
|
| 27 |
SIGLIP_MODEL_ID = "EYEDOL/siglipFULL-agri-finetuned"
|
| 28 |
+
LLAVA_MODEL_ID = "llava-hf/llava-1.5-7b-hf" # replace with actual model
|
| 29 |
DATASET_TEMPLATE = "EYEDOL/AGRILLAVA-image-text{}"
|
| 30 |
NUM_DATASETS = 1
|
| 31 |
BATCH_SIZE = 16
|