File size: 436 Bytes
25c2623
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from fastapi import FastAPI, File, UploadFile
from gradio_client import Client, handle_file

app = FastAPI()
client = Client("Myloiose/mobilenetv1-tflite-demo")

@app.post("/predict")
async def predict(file: UploadFile = File(...)):
    tmp = f"/tmp/{file.filename}"
    with open(tmp, "wb") as f:
        f.write(await file.read())
    result = client.predict(image=handle_file(tmp), api_name="/predict")
    return {"result": result}