首页 > 后端开发 > Python教程 > 使用 Gradio 和 Hugging Face 在 Lines 下使用 Python 代码构建文本提取器应用程序

使用 Gradio 和 Hugging Face 在 Lines 下使用 Python 代码构建文本提取器应用程序

Susan Sarandon
发布: 2024-11-01 11:20:02
原创
413 人浏览过

Build a Text Extractor App with Python Code Under Lines Using Gradio and Hugging Face

原帖:https://baxin.netlify.app/build-text-extractor-python-under-30-lines/

从图像中提取文本,称为光学字符识别 (OCR),对于文档处理、数据提取和可访问性应用程序来说是一项很有价值的功能。在本指南中,我们将使用 Python 库创建一个 OCR 应用程序,例如用于 OCR 的 pytesseract、用于图像处理的 Pillow 以及用于构建交互式 UI 的 Gradio。我们将在 Hugging Face Spaces 上部署此应用程序。

先决条件

开始之前,您需要一个 Hugging Face 帐户并对 Docker 有基本的了解。

分步指南

第 1 步:创建一个拥抱脸部空间

  1. 导航到 Hugging Face Spaces:登录 Hugging Face 并转到“Spaces”部分。
  2. 创建一个新空间
    • 点击“新空间”。
    • 为您的空间命名(例如,图像文本提取器)。
    • 选择Gradio作为SDK并设置可见性(公共或私有)。
    • 点击“创建空间”。

第 2 步:创建 Dockerfile

要在具有所需系统依赖项的 Hugging Face Spaces 上部署(例如用于 OCR 的 Tesseract),我们需要一个用于配置环境的 Dockerfile。

创建一个包含以下内容的 Dockerfile:

# Use an official Python runtime as a parent image
FROM python:3.12
ENV PIP_ROOT_USER_ACTION=ignore

# Set the working directory in the container
WORKDIR $HOME/app

# Install system dependencies
RUN apt-get update && apt-get install -y
RUN apt-get install -y tesseract-ocr
RUN apt-get install -y libtesseract-dev
RUN apt-get install -y libgl1-mesa-glx
RUN apt-get install -y libglib2.0-0
RUN pip install --upgrade pip

# Copy requirements and install dependencies
COPY requirements.txt requirements.txt
RUN pip install --no-cache-dir -r requirements.txt

# Copy the app code
COPY app.py ./

# Expose the port for Gradio
EXPOSE 7860

# Run the application
CMD ["python", "app.py"]
登录后复制

第 3 步:创建 OCR 应用程序

  1. 创建一个名为 app.py 的文件,其中包含以下内容:
import gradio as gr
import pytesseract
from PIL import Image
import os

def extract_text(image_path):
    if not image_path:
        return "No image uploaded. Please upload an image."

    if not os.path.exists(image_path):
        return f"Error: File not found at {image_path}"

    try:
        img = Image.open(image_path)
        text = pytesseract.image_to_string(img)
        return text if text.strip() else "No text detected in the image."
    except Exception as e:
        return f"An error occurred: {str(e)}"

iface = gr.Interface(
    fn=extract_text,
    inputs=gr.Image(type="filepath", label="Upload an image"),
    outputs=gr.Textbox(label="Extracted Text"),
    title="Image Text Extractor",
    description="Upload an image and extract text from it using OCR."
)

iface.launch(server_name="0.0.0.0", server_port=7860)
登录后复制
  1. 创建requirements.txt文件来指定依赖项:
gradio
pytesseract
Pillow
登录后复制

此设置包括:

  • 图像上传:gr.Image(type="filepath") 允许用户将图像作为文件路径上传,由 pytesseract 处理。
  • 文本提取:pytesseract.image_to_string 从图像中提取文本。
  • 用户界面:Gradio 生成一个简单的 UI,供用户上传图像和查看提取的文本。

第 4 步:将所有文件推送到拥抱面部空间

创建所有文件后,将它们推送到您的拥抱空间

以上是使用 Gradio 和 Hugging Face 在 Lines 下使用 Python 代码构建文本提取器应用程序的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:dev.to
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板