from fastapi import FastAPI, File, UploadFile from fastapi.staticfiles import StaticFiles from fastapi.responses import FileResponse from transformers import pipeline,AutoModel import streamlit as st from PIL import Image import torch import numpy as np import urllib.request import io import os from ofa_ocr import ofa_ocr_gr ##导入ofa_ocr中的大函数,方便使用其获取坐标的子函数 app = FastAPI() @app.get("/inference") def inference(): return "

Hello, World!

" @app.get("/infer_t5") def t5(input): return {"output": input} @app.post("/getCoordinates") async def getCoordinates(image: UploadFile = File(...)): img = await image.read() image = Image.open(io.BytesIO(img)).convert("L").convert("RGB") path_to_image = "save.jpg" image.save(path_to_image) ocr_api_func = ofa_ocr_gr() result = ocr_api_func(path_to_image) os.remove(path_to_image) return result