check prompt non empty

app.py

@@ -58,7 +58,9 @@ def start_over(gallery_state, loaded_model_setup):
     return gallery_state, None, None, gr.update(visible=False), loaded_model_setup
 
 def setup_model(prompt, model, seed, num_iterations, learning_rate, hps_w, imgrw_w, pcks_w, clip_w, progress=gr.Progress(track_tqdm=True)):
-
+    if prompt is None:
+        raise gr.Error("You forgot to provide a prompt !")
+    
     """Clear CUDA memory before starting the training."""
     torch.cuda.empty_cache()  # Free up cached memory
     
@@ -151,7 +153,7 @@ def generate_image(setup_args, num_iterations):
 
             if error_status["error_occurred"]:
                 torch.cuda.empty_cache()  # Free up cached memory
-                yield (None, "CUDA out of memory. Please reduce your batch size or image resolution.", None)
+                yield (None, "CUDA out of memory.", None)
             else:
                 main_thread.join()  # Ensure thread completion
                 final_image_path = os.path.join(save_dir, "best_image.png")