pythonImages / app.py
Myloiose's picture
Create app.py
25c2623 verified
raw
history blame
436 Bytes
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}