Update app.py

app.py

@@ -27,13 +27,13 @@ def set_system_prompt():
 # Function to enhance the prompt with Qwen model
 def enhance_prompt_with_qwen(prompt):
     client = Client("Qwen/Qwen2.5-72B-Instruct")
-    result = client.predict(
+    result,_,__ = client.predict(
         query=prompt,
         history=[],
         system="You are Qwen, an image generation prompt enhancer",
         api_name="/model_chat"
     )
-    return result['output']  # Assuming the enhanced prompt is under 'output'
+    return result # 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):