Update anime_model.py

anime_model.py

@@ -170,16 +170,19 @@ class Model:
         #if num_images > MAX_NUM_IMAGES:
         #    raise ValueError
         self.load("NormalBae")
+        model = NormalBaeDetector.from_pretrained(self.MODEL_ID)#.to("cuda")
+        torch.cuda.empty_cache()
+        gc.collect()
         control_image = self.preprocess(
             image=image,
             image_resolution=image_resolution,
             detect_resolution=preprocess_resolution,
         )
         if prompt == "":
-            prompt = "anime person"
+            prompt = "anime girl"
         print(prompt)
         return self.run_pipe(
-            prompt=self.get_prompt("Hentai Photo from imgur of " + prompt, additional_prompt),
+            prompt=self.get_prompt("Hentai Nude Anime Titties of " + prompt, additional_prompt),
             negative_prompt=negative_prompt,
             control_image=control_image,
             num_images=num_images,