zizytd commited on
Commit
feb2ea5
1 Parent(s): 8d4c960

updating the app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -132,8 +132,8 @@ def main():
132
  # st.error(f"Unexpected response format. Missing 'response_text' key. Received: {response.text}")
133
 
134
  if prompt := st.chat_input("What is up?"):
135
- response = pipe(f"<s>[INST] {prompt} [/INST]")
136
- response_text = response[0]["generated_text"].replace("<s>[INST]", "").replace("[/INST]", "").strip()
137
 
138
  with st.chat_message("assistant"):
139
  st.markdown(response_text)
 
132
  # st.error(f"Unexpected response format. Missing 'response_text' key. Received: {response.text}")
133
 
134
  if prompt := st.chat_input("What is up?"):
135
+ response = pipe(f"<s>[INST] {prompt} [/INST]")
136
+ response_text = response[0]["generated_text"].replace("<s>[INST]", "").replace("[/INST]", "").strip()
137
 
138
  with st.chat_message("assistant"):
139
  st.markdown(response_text)