Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoFeatureExtractor, VisionEncoderDecoderModel
 from PIL import Image
 import torch
-import spaces
+import spaces, time
 
 tokenizer = AutoTokenizer.from_pretrained("kha-white/manga-ocr-base")
 
@@ -26,7 +26,9 @@ def post_process(text):
 def manga_ocr(img):
   img = img.convert('L').convert('RGB')
   pixel_values = feature_extractor(img, return_tensors="pt").pixel_values.to("cuda")
+  start_time = time.time()
   output = model.generate(pixel_values)[0]
+  print("Time taken for OCR:", time.time() - start_time)
   text = tokenizer.decode(output, skip_special_tokens=True)
   text = post_process(text)
   return text