EYEDOL commited on
Commit
f30de13
·
verified ·
1 Parent(s): a48863e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "llava-hf/llava-1.5-7b-hf" # change if needed
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