amiguel commited on
Commit
1a046be
·
verified ·
1 Parent(s): dc99168

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -212,7 +212,7 @@ if prompt := st.chat_input("Ask your inspection question..."):
212
  if file_data["type"] == "table":
213
  predictions = classify_instruction(prompt, file_data["content"], model, tokenizer)
214
  result_df = file_data["content"][["Scope", "Functional Location", "Unit name"]].copy()
215
- result_df["Predicted Class"] = predictions
216
  st.write("Predicted Item Classes:")
217
  st.table(result_df)
218
  st.markdown(get_csv_download_link(result_df), unsafe_allow_html=True)
 
212
  if file_data["type"] == "table":
213
  predictions = classify_instruction(prompt, file_data["content"], model, tokenizer)
214
  result_df = file_data["content"][["Scope", "Functional Location", "Unit name"]].copy()
215
+ result_df["Item Class"] = predictions
216
  st.write("Predicted Item Classes:")
217
  st.table(result_df)
218
  st.markdown(get_csv_download_link(result_df), unsafe_allow_html=True)