Fix error in generate

app.py

@@ -45,11 +45,11 @@ def train_model(*inputs):
     args_general = argparse.Namespace(
                 image_captions_filename = True,
                 train_text_encoder = True,
-                stop_text_encoder_training = stptxt,
+                #stop_text_encoder_training = stptxt,
                 save_n_steps = 0,
                 pretrained_model_name_or_path = model_to_load,
                 instance_data_dir="instance_images",
-                class_data_dir=class_data_dir,
+                #class_data_dir=class_data_dir,
                 output_dir="output_model",
                 instance_prompt="",
                 seed=42,
@@ -61,7 +61,7 @@ def train_model(*inputs):
                 learning_rate=2e-6,
                 lr_scheduler="polynomial",
                 lr_warmup_steps = 0,
-                max_train_steps=Training_Steps,     
+                #max_train_steps=Training_Steps,     
     )
     run_training(args_general)
     torch.cuda.empty_cache()