gryan-galario commited on
Commit
8bfffa0
1 Parent(s): b7694dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = self.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)
 
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)