Spaces:
Running
Running
updating the app
Browse files
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 |
-
|
136 |
-
|
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)
|