Update model.py

model.py

@@ -117,6 +117,16 @@ class Model:
     def generate(
         self, editing_type: str, hairstyle_index: int, color_description: str, latent: torch.Tensor
     ) -> np.ndarray:
+        # Giả sử dataset được tạo hoặc lấy từ một phương thức hoặc thuộc tính nào đó
+        dataset = self.prepare_dataset(editing_type, hairstyle_index, color_description, latent)
+        if dataset is None or not dataset:
+            raise ValueError("Dataset không được tạo hoặc là rỗng")
+        if dataset[0] is None:
+            raise ValueError("Phần tử đầu tiên của dataset là None, không thể tiếp tục")
+        
+        w, hairstyle_text_inputs_list, color_text_inputs_list = dataset[0][:3]
+        # Tiếp tục xử lý với w, hairstyle_text_inputs_list, và color_text_inputs_list
+
         opts = self.hairclip.opts
         opts.editing_type = editing_type
         opts.color_description = color_description