Update app.py

app.py

@@ -14,6 +14,17 @@ API_URL_DEV = "https://api-inference.huggingface.co/models/black-forest-labs/FLU
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 timeout = 100
 
+# Function to set the system prompt once
+def set_system_prompt():
+    client = Client("Qwen/Qwen2.5-72B-Instruct")
+    result = client.predict(
+        system="You are Qwen, an image generation prompt enhancer",
+        api_name="/modify_system_session"
+    )
+    print(f"System session modified: {result}")
+    return result
+
+# Function to enhance the prompt with Qwen model
 def enhance_prompt_with_qwen(prompt):
     client = Client("Qwen/Qwen2.5-72B-Instruct")
     result = client.predict(
@@ -24,7 +35,11 @@ def enhance_prompt_with_qwen(prompt):
     )
     return result['output']  # Assuming the enhanced prompt is under 'output'
 
+# Image generation query function
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, huggingface_api_key=None, use_dev=False):
+    # Set system prompt first
+    set_system_prompt()
+
     # Enhance the prompt before translation
     enhanced_prompt = enhance_prompt_with_qwen(prompt)