ydshieh commited on
Commit ·
6d11b0d
1
Parent(s): b9b6785
update after integration
Browse files
md.py
CHANGED
|
@@ -11,7 +11,7 @@ model = Kosmos2_5ForConditionalGeneration.from_pretrained(repo, device_map=devic
|
|
| 11 |
processor = AutoProcessor.from_pretrained(repo)
|
| 12 |
|
| 13 |
# sample image
|
| 14 |
-
url = "https://huggingface.co/microsoft/kosmos-2.5/
|
| 15 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 16 |
|
| 17 |
prompt = "<md>"
|
|
@@ -30,4 +30,4 @@ generated_ids = model.generate(
|
|
| 30 |
)
|
| 31 |
|
| 32 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
| 33 |
-
print(generated_text[0])
|
|
|
|
| 11 |
processor = AutoProcessor.from_pretrained(repo)
|
| 12 |
|
| 13 |
# sample image
|
| 14 |
+
url = "https://huggingface.co/microsoft/kosmos-2.5/resolve/main/receipt_00008.png"
|
| 15 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 16 |
|
| 17 |
prompt = "<md>"
|
|
|
|
| 30 |
)
|
| 31 |
|
| 32 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)
|
| 33 |
+
print(generated_text[0])
|
ocr.py
CHANGED
|
@@ -11,7 +11,7 @@ model = Kosmos2_5ForConditionalGeneration.from_pretrained(repo, device_map=devic
|
|
| 11 |
processor = AutoProcessor.from_pretrained(repo)
|
| 12 |
|
| 13 |
# sample image
|
| 14 |
-
url = "https://huggingface.co/microsoft/kosmos-2.5/
|
| 15 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 16 |
|
| 17 |
# bs = 1
|
|
@@ -70,4 +70,4 @@ for line in lines:
|
|
| 70 |
continue
|
| 71 |
line = list(map(int, line[:8]))
|
| 72 |
draw.polygon(line, outline="red")
|
| 73 |
-
image.save("output.png")
|
|
|
|
| 11 |
processor = AutoProcessor.from_pretrained(repo)
|
| 12 |
|
| 13 |
# sample image
|
| 14 |
+
url = "https://huggingface.co/microsoft/kosmos-2.5/resolve/main/receipt_00008.png"
|
| 15 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 16 |
|
| 17 |
# bs = 1
|
|
|
|
| 70 |
continue
|
| 71 |
line = list(map(int, line[:8]))
|
| 72 |
draw.polygon(line, outline="red")
|
| 73 |
+
image.save("output.png")
|