Create api.py

api.py

@@ -0,0 +1,31 @@
+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 "获取图片保存成功"