fixing training

app.py

@@ -226,13 +226,17 @@ class Demo:
         outputs=[self.train_button,  self.train_status, self.download, self.model_dropdown]
         )
 
-    def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, train_method, neg_guidance, iterations, lr, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
+    def train(self, target_concept,positive_prompt, negative_prompt, rank, iterations_input, lr_input, attributes_input, is_person, pbar = gr.Progress(track_tqdm=True)):
         if target_concept is None:
             target_concept = ''
         if positive_prompt is None:
             positive_prompt = ''
         if negative_prompt is None:
             negative_prompt = ''
+        if is_person is None:
+            is_person = False
+        else:
+            is_person = True
         print(target_concept, positive_prompt, negative_prompt, attributes_input, is_person)
         
         randn = torch.randint(1, 10000000, (1,)).item()