Upload infer_gradio.py

infer_gradio.py

@@ -0,0 +1,25 @@
+from manga_ocr import MangaOcr
+from PIL import Image
+import time, random, io, base64
+import gradio as gr
+import spaces
+
+mocr = MangaOcr()
+
+@spaces.GPU
+def ocr_text(image):
+    start = time.time()
+    text = mocr(image)
+    print(time.time() - start)
+    return text
+
+iface = gr.Interface(
+    fn=ocr_text,
+    inputs=gr.Image(type='pil'),
+    outputs="text",
+    title="Manga OCR",
+    description="Extrait le texte d'une image de manga.",
+)
+
+if __name__ == '__main__':
+    iface.launch(server_name="0.0.0.0", server_port=9999)