Update app.py

app.py

@@ -382,14 +382,14 @@ def llava_bot(state, temperature, top_p, max_new_tokens, request: gr.Request):
 
 
 # UI Components
-title_markdown = """
+title_markdown = + f'Loaded Model: {DEFAULT_MODEL}' + """
 # LLAVAGUARD: VLM-based Safeguard for Vision Dataset Curation and Safety Assessment
 [[Project Page](https://ml-research.github.io/human-centered-genai/projects/llavaguard/index.html)]
 [[Code](https://github.com/ml-research/LlavaGuard)]
 [[Model](https://huggingface.co/collections/AIML-TUDA/llavaguard-665b42e89803408ee8ec1086)]
 [[Dataset](https://huggingface.co/datasets/aiml-tuda/llavaguard)]
 [[LavaGuard](https://arxiv.org/abs/2406.05113)]
-"""
+""" 
 
 tos_markdown = """
 ### Terms of use