diff --git a/modules/api/api.py b/modules/api/api.py
index 518b2a61..8134fe58 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -6,7 +6,7 @@ import uvicorn
 import gradio as gr
 from threading import Lock
 from io import BytesIO
-from gradio.processing_utils import decode_base64_to_file
+from gradio_client.processing_utils import decode_base64_to_file
 from fastapi import APIRouter, Depends, FastAPI, Request, Response
 from fastapi.security import HTTPBasic, HTTPBasicCredentials
 from fastapi.exceptions import HTTPException