ninjals commited on
Commit
dc74efa
·
verified ·
1 Parent(s): 543fa40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline
6
 
7
  BASE_MODEL_ID = "HuggingFaceTB/SmolVLM2-500M-Video-Instruct"
8
  FINE_TUNED_MODEL_ID = "ninjals/FoodExtract-Vision-SmolVLM2-500M-fine-tune-v1-VIDEO"
9
- OUTPUT_TOKENS = 128
10
 
11
  # Load original base model (no fine-tuning)
12
  print(f"[INFO] Loading Original Model")
 
6
 
7
  BASE_MODEL_ID = "HuggingFaceTB/SmolVLM2-500M-Video-Instruct"
8
  FINE_TUNED_MODEL_ID = "ninjals/FoodExtract-Vision-SmolVLM2-500M-fine-tune-v1-VIDEO"
9
+ OUTPUT_TOKENS = 256
10
 
11
  # Load original base model (no fine-tuning)
12
  print(f"[INFO] Loading Original Model")