Spaces:
Runtime error
Runtime error
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}
|