AskUI
/

PTA-1

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

maxiw commited on 4 days ago

Commit

767c22e

•

1 Parent(s): 4f71621

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ processor = AutoProcessor.from_pretrained("AskUI/PTA-1", trust_remote_code=True)
 task_prompt = "<OPEN_VOCABULARY_DETECTION>"
 prompt = task_prompt + "description of the target element"
-image = Image.open("path to screenshot")
 inputs = processor(text=prompt, images=image, return_tensors="pt").to(device, torch_dtype)

 task_prompt = "<OPEN_VOCABULARY_DETECTION>"
 prompt = task_prompt + "description of the target element"
+image = Image.open("path to screenshot").convert("RGB")
 inputs = processor(text=prompt, images=image, return_tensors="pt").to(device, torch_dtype)