Spaces:
Running
Running
gryan-galario
commited on
Commit
•
8bfffa0
1
Parent(s):
b7694dc
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def post_process(text):
|
|
20 |
|
21 |
def manga_ocr(img):
|
22 |
img = img.convert('L').convert('RGB')
|
23 |
-
pixel_values =
|
24 |
output = model.generate(pixel_values)[0]
|
25 |
text = tokenizer.decode(output, skip_special_tokens=True)
|
26 |
text = post_process(text)
|
|
|
20 |
|
21 |
def manga_ocr(img):
|
22 |
img = img.convert('L').convert('RGB')
|
23 |
+
pixel_values = feature_extractor(img, return_tensors="pt").pixel_values
|
24 |
output = model.generate(pixel_values)[0]
|
25 |
text = tokenizer.decode(output, skip_special_tokens=True)
|
26 |
text = post_process(text)
|