add script for generating dataset

This commit is contained in:
2025-01-12 00:42:14 +00:00
parent 14c6f26ddc
commit d2b6e70734
4 changed files with 164 additions and 13 deletions

View File

@@ -23,7 +23,7 @@ pathlib.Path("./samples").mkdir(parents=True, exist_ok=True)
img = Image.open("samples/Untitled.jpg")
md_answer = moondream.answer_question(
moondream.encode_image(img),
"Describe this image.",
"Is this image AI generated?",
tokenizer=tokenizer,
num_beams=4,
no_repeat_ngram_size=5,