Spaces:
Running
Running
final app change
Browse files
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 |
|