Update app.py

app.py

@@ -40,7 +40,7 @@ def ask_assistant_stream(st_model, st_messages, st_temp_value, st_max_tokens):
             stream=True,
             max_tokens=st_max_tokens,
         )
-        response["stream"]=stream
+        response["stream"] = stream
     
     except Exception as e:
         pass
@@ -142,6 +142,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     
     # Display assistant response in chat message container
     assistant = ask_assistant_stream(model_links[selected_model], st.session_state.messages, temp_values, max_token_value)
+    st.write assistant
     if "stream" in assistant: 
         with st.chat_message("assistant"):
             response = st.write_stream(assistant["stream"])