Linhz commited on
Commit
8ebe27a
·
verified ·
1 Parent(s): 204afe5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -9,6 +9,9 @@ import torch
9
 
10
  st.set_page_config(page_title = "Vietnamese Legal Question Answering System", page_icon= "🐧", layout="centered", initial_sidebar_state="collapsed")
11
 
 
 
 
12
  st.markdown(
13
  """
14
  <h1 style="text-align: center;">Vietnamese Legal Question Answering System</h1>
@@ -73,11 +76,11 @@ for message in st.session_state.messages:
73
  if message['role'] == 'assistant':
74
  avatar_class = "assistant-avatar"
75
  message_class = "assistant-message"
76
- avatar = './static/AI.png'
77
  else:
78
  avatar_class = "user-avatar"
79
  message_class = "user-message"
80
- avatar = './static/human.jpg'
81
  st.markdown(f"""
82
  <div class="{message_class}">
83
  <img src="{avatar}" class="{avatar_class}" />
 
9
 
10
  st.set_page_config(page_title = "Vietnamese Legal Question Answering System", page_icon= "🐧", layout="centered", initial_sidebar_state="collapsed")
11
 
12
+ with open("./static/styles.css") as f:
13
+ st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
14
+
15
  st.markdown(
16
  """
17
  <h1 style="text-align: center;">Vietnamese Legal Question Answering System</h1>
 
76
  if message['role'] == 'assistant':
77
  avatar_class = "assistant-avatar"
78
  message_class = "assistant-message"
79
+ avatar = './app/static/AI.png'
80
  else:
81
  avatar_class = "user-avatar"
82
  message_class = "user-message"
83
+ avatar = './app/static/human.jpg'
84
  st.markdown(f"""
85
  <div class="{message_class}">
86
  <img src="{avatar}" class="{avatar_class}" />