Update app.py

app.py

@@ -56,8 +56,7 @@ def promptifier(prompt):
     system_instructions1 = "<s>[SYSTEM] Your task is to modify prompt by USER to more better prompt for Image Generation in Stable Diffusion XL, reply with prompt only, Your task is to reply with final prompt in SDXL image generation format only. [USER]"
     formatted_prompt = f"{system_instructions1} {prompt} [FINAL_PROMPT]"
     stream = client1.text_generation(formatted_prompt, max_new_tokens=80, stream=True, details=True, return_full_text=False)
-    instructions = "".join([response.token.text for response in stream if response.token.text != "</s>"])
-    return instructions
+    return "".join([response.token.text for response in stream if response.token.text != "</s>"])
 
 # Generator
 @spaces.GPU(duration=30, queue=False)
@@ -111,8 +110,9 @@ def king(type ,
         else:
             guidance_scale2=(guidance_scale/2) 
             if enhance_prompt:
+                print(f"BEFORE: {instruction} ")
                 instruction = promptifier(instruction)
-                
+                print(f"AFTER: {instruction} ")
             image = pipe( prompt = instruction,
             negative_prompt=negative_prompt,
             guidance_scale = guidance_scale2,