update

chatbot.py

@@ -1,6 +1,7 @@
-import torch
-
+import os
+import sys
 
+import torch
 from model import IntentPredictModel
 from transformers import T5Tokenizer, GPT2LMHeadModel, GPT2Tokenizer
 from diffusers import StableDiffusionPipeline
@@ -28,6 +29,9 @@ class Chat:
         self.human_words = ["man", "men", "woman", "women", "people", "person", "human", "male", "female", "boy", "girl", "child", "kid", "baby", "player"]
         self.negative_prompt="cartoon, anime, ugly, asian, (aged, white beard, black skin, wrinkle:1.1), (bad proportions, unnatural feature, incongruous feature:1.4), (blurry, un-sharp, fuzzy, un-detailed skin:1.2), (facial contortion, poorly drawn face, deformed iris, deformed pupils:1.3), (mutated hands and fingers:1.5), disconnected hands, disconnected limbs"
         
+        self.save_images_folder = os.path.join(sys.path[0], "generated_images")
+        os.makedirs(self.save_images_folder, exist_ok=True)
+        
         self.context_for_intent = ""
         self.context_for_text_dialog = ""
     
@@ -112,7 +116,8 @@ class Chat:
                 num_inference_steps=20,
                 guidance_scale=7.5,
                 generator=generator).images[0]
-            save_image_path = f"generated_images/{response}.png"
+            
+            save_image_path = f"{self.save_images_folder}/{response}.png"
             image.save(save_image_path)
 
             self.context_for_intent += " [SEP] " + response