Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModel, AutoFeatureExtractor
+from transformers import AutoTokenizer, AutoModel, VisionEncoderDecoderModel
 from PIL import Image
 import requests
 import torch
@@ -7,7 +7,7 @@ import torch
 
 tokenizer = AutoTokenizer.from_pretrained("kha-white/manga-ocr-base")
 
-model = AutoModel.from_pretrained("kha-white/manga-ocr-base")
+model = VisionEncoderDecoderModel.from_pretrained("kha-white/manga-ocr-base")
 
 feature_extractor = AutoFeatureExtractor.from_pretrained("kha-white/manga-ocr-base")
 
@@ -28,7 +28,7 @@ def manga_ocr(img):
   return text
 
 iface = gr.Interface(
-    fn=infer,
+    fn=manga_ocr,
     inputs=[gr.inputs.Image(label="Input", type="pil")],
     outputs="text",
     layout="horizontal",