mrdbourke/FoodExtract-1k-Vision
Viewer • Updated • 1.51k • 84 • 6
How to use sergiudanstan/smolvlm2-2.2b-foodextract-lora with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolVLM2-2.2B-Instruct")
model = PeftModel.from_pretrained(base_model, "sergiudanstan/smolvlm2-2.2b-foodextract-lora")LoRA adapter for HuggingFaceTB/SmolVLM2-2.2B-Instruct,
fine-tuned on mrdbourke/FoodExtract-1k-Vision.
Given a food/drink image it returns structured JSON:
{"is_food": 1, "image_title": "salad",
"food_items": ["tomato", "lettuce"], "drink_items": []}
from transformers import AutoProcessor, AutoModelForImageTextToText
from peft import PeftModel
base = "HuggingFaceTB/SmolVLM2-2.2B-Instruct"
processor = AutoProcessor.from_pretrained(base)
model = AutoModelForImageTextToText.from_pretrained(base)
model = PeftModel.from_pretrained(model, "sergiudanstan/smolvlm2-2.2b-foodextract-lora").merge_and_unload()
Base model
HuggingFaceTB/SmolLM2-1.7B