Spaces:
Runtime error
Runtime error
| from fastapi import FastAPI, File, UploadFile | |
| from gradio_client import Client, handle_file | |
| app = FastAPI() | |
| client = Client("Myloiose/mobilenetv1-tflite-demo") | |
| 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} | |