Update app.py

app.py

@@ -62,6 +62,7 @@ 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})')
 
 
 css = """