Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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["
|
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)
|