app.py

# Gradioのインポート
import gradio as gr
import subprocess

# subprocessモジュールを使用してpip installコマンドを実行
subprocess.run(["pip", "install", "doctr"])
subprocess.run(["pip", "install", "doctr.io"])
# cn Gradioのインターフェースを追加する前に必要なライブラリをインポート
from doctr.io import DocumentFile
from doctr.models import ocr_predictor, from_hub

# OCRの予測を行う関数を定義
def predict_ocr(image_path):
    img = DocumentFile.from_images([image_path])
    res = predictor(img)
    return res

# 画像をアップロードする入力コンポーネントを作成
image_input = gr.inputs.Image(upload=True)

# 結果を表示する出力コンポーネントを作成
output_text = gr.outputs.Textbox(label="OCR Result")

# Gradioのインターフェースを作成して起動
gr.Interface(fn=predict_ocr, inputs=image_input, outputs=output_text, title="OCR Predictor").launch()