Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -146,7 +146,7 @@ if prompt := st.chat_input("Ask your inspection question..."):
|
|
146 |
try:
|
147 |
with st.chat_message("assistant", avatar=BOT_AVATAR):
|
148 |
predicted_class = classify_instruction(prompt, file_context, model, tokenizer)
|
149 |
-
response = f"
|
150 |
st.markdown(response)
|
151 |
st.session_state.messages.append({"role": "assistant", "content": response})
|
152 |
|
|
|
146 |
try:
|
147 |
with st.chat_message("assistant", avatar=BOT_AVATAR):
|
148 |
predicted_class = classify_instruction(prompt, file_context, model, tokenizer)
|
149 |
+
response = f"The Item Class is: {predicted_class}"
|
150 |
st.markdown(response)
|
151 |
st.session_state.messages.append({"role": "assistant", "content": response})
|
152 |
|