diff --git a/nexa/cli/entry.py b/nexa/cli/entry.py
index 66c3ca70..d8bdddb5 100644
--- a/nexa/cli/entry.py
+++ b/nexa/cli/entry.py
@@ -51,9 +51,9 @@ def run_ggml_inference(args):
         if hasattr(args, 'streamlit') and args.streamlit:
             inference.run_streamlit(model_path)
         elif args.img2img:
-            inference.loop_img2img()
+            inference.run_img2img()
         else:
-            inference.loop_txt2img()
+            inference.run_txt2img()
         return
     elif args.command == "vlm":
         from nexa.gguf.nexa_inference_vlm import NexaVLMInference
diff --git a/nexa/onnx/nexa_inference_image.py b/nexa/onnx/nexa_inference_image.py
index 87d1f959..4abb82fe 100644
--- a/nexa/onnx/nexa_inference_image.py
+++ b/nexa/onnx/nexa_inference_image.py
@@ -31,8 +31,8 @@ class NexaImageInference:
     A class used for loading image models and running image generation.
 
     Methods:
-    run: Run the image generation loop.
-    run_streamlit: Run the Streamlit UI.
+        run: Run the image generation loop.
+        run_streamlit: Run the Streamlit UI.
 
     Args:
     model_path (str): Path or identifier for the model in Nexa Model Hub.