Update app.py

app.py

@@ -72,7 +72,7 @@ def handle_chat(user_message):
             img.save(buffered, format="PNG")
             img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-            img = f'<img src="data:image/png;base64,{img_str}" style="width:600px; height:500px;">'
+            img = f'<img src="data:image/png;base64,{img_str}" style="width:600px; height:600px;">'
 
             ai_response =   img + ai_response