amiguel commited on
Commit
33be14e
·
verified ·
1 Parent(s): 004fb60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"Predicted class: {predicted_class}"
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