Update app.py

app.py

@@ -112,7 +112,7 @@ def generate_response(prompt):
 
             response = response.json()
             imageurl = response['data'][0]['url']
-            ie = f"Here is the generated image:\n {imageurl} \n \n Model used: `{model_name}`"
+            ie = f"Here is the generated image:\n {imageurl} \n \n Model used: `{image_model}`"
             st.session_state['messages'].append({"role": "assistant", "content": ie})
 
         except requests.exceptions.JSONDecodeError as e: