albhu commited on
Commit
d60519f
·
verified ·
1 Parent(s): daf0121

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -9,13 +9,9 @@ st.title("Jai")
9
  if "previous_conversations" not in st.session_state:
10
  st.session_state.previous_conversations = []
11
 
12
- # Korábbi beszélgetések törlése
13
- if st.checkbox("Korábbi Beszélgetések Törlése"):
14
- st.session_state.previous_conversations = []
15
-
16
  # Tokenizáló modell és válaszgeneráló modell választása
17
- tokenizer_name = st.selectbox("Tokenizáló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
18
- generator_name = st.selectbox("Válaszgeneráló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
19
 
20
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
21
  model = AutoModelForCausalLM.from_pretrained(generator_name, trust_remote_code=True)
@@ -81,3 +77,9 @@ if document_file is not None:
81
 
82
  # Aktuális beszélgetés hozzáadása az előző beszélgetésekhez
83
  st.session_state.previous_conversations.append({"input_text": input_text, "response": response})
 
 
 
 
 
 
 
9
  if "previous_conversations" not in st.session_state:
10
  st.session_state.previous_conversations = []
11
 
 
 
 
 
12
  # Tokenizáló modell és válaszgeneráló modell választása
13
+ tokenizer_name = st.sidebar.selectbox("Tokenizáló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
14
+ generator_name = st.sidebar.selectbox("Válaszgeneráló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
15
 
16
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
17
  model = AutoModelForCausalLM.from_pretrained(generator_name, trust_remote_code=True)
 
77
 
78
  # Aktuális beszélgetés hozzáadása az előző beszélgetésekhez
79
  st.session_state.previous_conversations.append({"input_text": input_text, "response": response})
80
+
81
+ # Előző beszélgetések megjelenítése
82
+ st.sidebar.subheader("Előző Beszélgetések")
83
+ for conv in st.session_state.previous_conversations:
84
+ st.sidebar.write(conv["input_text"])
85
+ st.sidebar.write(conv["response"])