diff --git a/api/launch-extras.bat b/api/launch-extras.bat
index 2f0b95c0..126df2c6 100644
--- a/api/launch-extras.bat
+++ b/api/launch-extras.bat
@@ -3,4 +3,4 @@ IF "%ONNX_WEB_EXTRA_MODELS%"=="" (set ONNX_WEB_EXTRA_MODELS=extras.json)
 python -m onnx_web.convert --sources --diffusion --upscaling --correction --extras=%ONNX_WEB_EXTRA_MODELS% --token=%HF_TOKEN%
 
 echo "Launching API server..."
-flask --app="onnx_web.serve:run" run --host=0.0.0.0
+flask --app="onnx_web.main:run" run --host=0.0.0.0
diff --git a/api/launch.bat b/api/launch.bat
index 4ee27fd6..09ddfccd 100644
--- a/api/launch.bat
+++ b/api/launch.bat
@@ -2,4 +2,4 @@ echo "Downloading and converting models to ONNX format..."
 python -m onnx_web.convert --sources --diffusion --upscaling --correction --token=%HF_TOKEN%
 
 echo "Launching API server..."
-flask --app="onnx_web.serve:run" run --host=0.0.0.0
+flask --app="onnx_web.main:run" run --host=0.0.0.0