Faysal4200 commited on
Commit
8fb8ba5
·
verified ·
1 Parent(s): 6a03916

final app change

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -470,7 +470,7 @@ if st.button("Generate VLM Explanation"):
470
  vlm_info["model"],
471
  vlm_info["device"],
472
  img_for_vlm,
473
- st.session_state["pred_label"],
474
  max_new_tokens=128
475
  )
476
 
 
470
  vlm_info["model"],
471
  vlm_info["device"],
472
  img_for_vlm,
473
+ full_names[st.session_state["pred_label"]],
474
  max_new_tokens=128
475
  )
476