Update app.py

app.py

@@ -12,7 +12,6 @@ model = VisionEncoderDecoderModel.from_pretrained("kha-white/manga-ocr-base")
 model.to("cuda")
 
 feature_extractor = AutoFeatureExtractor.from_pretrained("kha-white/manga-ocr-base")
-feature_extractor.to("cuda")
 
 examples = ["00.jpg", "01.jpg", "02.jpg", "03.jpg", "04.jpg", "05.jpg", "06.jpg", "07.jpg", "08.jpg", "09.jpg", "10.jpg", "11.jpg"]
 
@@ -26,7 +25,7 @@ def post_process(text):
 @spaces.GPU
 def manga_ocr(img):
   img = img.convert('L').convert('RGB')
-  pixel_values = feature_extractor(img, return_tensors="pt").pixel_values
+  pixel_values = feature_extractor(img, return_tensors="pt").pixel_values.to("cuda")
   output = model.generate(pixel_values)[0]
   text = tokenizer.decode(output, skip_special_tokens=True)
   text = post_process(text)