|
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 |
|
|
|
app = FastAPI() |
|
|
|
@app.get("/inference") |
|
def inference(): |
|
return "<p>Hello, World!</p>" |
|
|
|
@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) |
|
return "获取图片保存成功" |
