Update app.py

app.py

@@ -147,12 +147,13 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     if selected_model == "Nous-Hermes-2-Yi-34B":
         stream = interact_with_together_api(st.session_state.messages)
     else:
-        interact_with_huggingface_model(st.session_state.messages, model_links[selected_model])
+        stream = interact_with_huggingface_model(st.session_state.messages, model_links[selected_model])
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         response = ""
-        for chunk in stream:
-            response = chunk
-            st.markdown(response)
+        if stream:
+            for chunk in stream:
+                response = chunk
+                st.markdown(response)
     st.session_state.messages.append(("assistant", response))