hugginngface no acepta write_stream

app.py

@@ -5,9 +5,7 @@ import numpy as np
 import pandas as pd
 import streamlit as st
 
-def stream_data(prompt):
-    for _ in model(prompt,_stream):
-        yield str(_) 
+
 
 if "model" not in st.session_state:
     st.session_state["model"] = "TheBloke/Mistral-7B-OpenOrca-GGUF"
@@ -30,18 +28,8 @@ if prompt := st.chat_input("Pregúntale a Don Simón"):
         st.markdown(prompt)
 
 with st.chat_message("assistant"):
-    stream = model("Hola",max_new_tokens=25, stream=True)
-    response = st.write_stream(stream)
-    #st.session_state.messages.append({"role": "assistant", "content": response})
-
+    stream = model(prompt,max_new_tokens=10)
+    response = st.write(stream)
+    st.session_state.messages.append({"role": "assistant", "content": response})
 
-'''
-text= st.text_area("Pregúntale a Don Simón")
 
-if text:
-    #answer = similarity_search(text,5)
-    answer=""
-    for _ in model(text,max_new_tokens=25, stream=True):
-        answer= answer+str(_)
-        print(answer)'''
-