Update feifeilib/feifeiprompt.py

feifeilib/feifeiprompt.py

@@ -1,11 +1,9 @@
 import numpy as np
 import random
-#from extras.expansion import FooocusExpansion
 import re
 import torch
 import config
 
-#expansion = FooocusExpansion()
 with open("artist.txt", "r") as file:
     artists = file.readlines()
 
@@ -45,12 +43,9 @@ def feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusE
                     prompt += style["prompt"].replace("{prompt}", "the ")
 
     if FooocusExpansion_select:
-        # prompt = expansion(prompt, seed)
-        prompt = prompt
-    # prompt = f"feifei, flux, real model girl in real life, {prompt}, slight smile, Master of Light and Shadow."
-    prompt = re.sub("girl", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
-    prompt = re.sub("young woman", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
-    prompt = re.sub("woman", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
-    prompt = re.sub("model", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
+        prompt = re.sub("girl", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
+        prompt = re.sub("young woman", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
+        prompt = re.sub("woman", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
+        prompt = re.sub("model", " feifei, A beautiful, 18 yo kpop idol, large-busted Japanese girl, with light makeup, gazing deeply into the camera, " ,prompt)
      
     return prompt,generator