EYEDOL commited on
Commit
bcda1e2
·
verified ·
1 Parent(s): 7679f34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "your-llava-model-hf-id" # replace with actual model
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