Update app.py
Browse files
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"])
|