SmolVLM

Running on Zero

merve HF staff commited on Aug 6

Commit

9ad1205

•

1 Parent(s): e352103

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/idefics3
 BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
 EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
-#@spaces.GPU
 def model_inference(
     images, text, decoding_strategy, temperature, max_new_tokens,
     repetition_penalty, top_p

 BAD_WORDS_IDS = processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
 EOS_WORDS_IDS = [processor.tokenizer.eos_token_id]
+@spaces.GPU
 def model_inference(
     images, text, decoding_strategy, temperature, max_new_tokens,
     repetition_penalty, top_p