Update app.py

app.py

@@ -61,8 +61,8 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
 
     image_bytes = requests.post(API_URL, headers=headers, json=payload).content
     image = Image.open(io.BytesIO(image_bytes))
-    return image
     print(f'\033[1mГенерация {key} завершена!\033[0m ({prompt})')
+    return image
 
 
 css = """