Fix tkinter issue for headless

bmaltais · Apr 15, 2024 · f4658f9 · f4658f9
1 parent e519ffe
commit f4658f9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/kohya_gui/dreambooth_folder_creation_gui.py b/kohya_gui/dreambooth_folder_creation_gui.py
@@ -1,5 +1,4 @@
 import gradio as gr
-from easygui import msgbox
 from .common_gui import get_folder_path, scriptdir, list_dirs, create_refresh_button
 import shutil
 import os
@@ -45,12 +44,12 @@ def dreambooth_folder_preparation(
 
     # Check for instance prompt
     if util_instance_prompt_input == "":
-        msgbox("Instance prompt missing...")
+        log.error("Instance prompt missing...")
         return
 
     # Check for class prompt
     if util_class_prompt_input == "":
-        msgbox("Class prompt missing...")
+        log.error("Class prompt missing...")
         return
 
     # Create the training_dir path