|
|
import gradio as gr |
|
|
import pytesseract |
|
|
import cv2 |
|
|
import numpy as np |
|
|
from PIL import Image |
|
|
from io import BytesIO |
|
|
|
|
|
def extract_text(image): |
|
|
image = Image.open(image.name) |
|
|
image = image.convert('L') |
|
|
image = np.array(image) |
|
|
text = pytesseract.image_to_string(image) |
|
|
print(text) |
|
|
return text |
|
|
|
|
|
|
|
|
inputs = gr.Image(label='Upload an image',image_mode='L',tool='select',type='pil') |
|
|
outputs = gr.outputs.Textbox() |
|
|
interface = gr.Interface(fn=extract_text, inputs=inputs, outputs=outputs, |
|
|
title="Extraction de texte à partir d'une image", |
|
|
description="Téléchargez une image contenant du texte et cliquez sur 'Predict' pour extraire le texte.") |
|
|
|
|
|
interface.launch() |
|
|
|
