Spaces:

prithivMLmods
/

Tiny-VLMs-Lab

Running on Zero

prithivMLmods commited on 10 days ago

Commit

eb72331

verified ·

1 Parent(s): 70a54c4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -117,7 +117,7 @@ moondream = AutoModelForCausalLM.from_pretrained(
 tokenizer_md = AutoTokenizer.from_pretrained(MODEL_ID_MD, revision=REVISION_MD)
 # --- SmolVLM2 Model Loading ---
-MODEL_ID_S2 = "HuggingFaceTB/SmolVLM2-2.2B-Instruct"
 processor_s2 = AutoProcessor.from_pretrained(MODEL_ID_S2, trust_remote_code=True)
 model_s2 = AutoModelForImageTextToText.from_pretrained(
     MODEL_ID_S2,

 tokenizer_md = AutoTokenizer.from_pretrained(MODEL_ID_MD, revision=REVISION_MD)
 # --- SmolVLM2 Model Loading ---
+MODEL_ID_S2 = "HuggingFaceTB/SmolVLM2-256M-Video-Instruct"
 processor_s2 = AutoProcessor.from_pretrained(MODEL_ID_S2, trust_remote_code=True)
 model_s2 = AutoModelForImageTextToText.from_pretrained(
     MODEL_ID_S2,