diff --git a/main.py b/main.py
index fd88804..3314d9f 100644
--- a/main.py
+++ b/main.py
@@ -40,6 +40,7 @@ from utilities.logger import Logger
 from utilities.model import Model
 from utilities.config import Config
 from utilities.text2img import Text2Img
+from utilities.img2img import Img2Img
 
 
 app = Flask(__name__)