Spaces:
Sleeping
Sleeping
sandrocalzada
commited on
Commit
•
0df2697
1
Parent(s):
2880c2f
Update app.py
Browse files
app.py
CHANGED
@@ -91,27 +91,16 @@ with st.sidebar:
|
|
91 |
st.success("Done")
|
92 |
|
93 |
# Model selection buttons
|
94 |
-
st.subheader("Select Model")
|
|
|
95 |
col1, col2 = st.columns(2)
|
96 |
-
if
|
97 |
-
st.session_state.
|
98 |
-
|
99 |
-
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
with col2:
|
104 |
-
if st.button("Gemini", key="Gemini",
|
105 |
-
on_click=lambda: setattr(st.session_state, 'selected_model', 'Gemini')):
|
106 |
-
st.session_state.selected_model = 'Gemini'
|
107 |
-
|
108 |
-
# Apply the selected styling
|
109 |
-
if st.session_state.selected_model == 'Gemma':
|
110 |
-
col1.button("Gemma", key="reGemma", on_click=None, disabled=True, style="background-color: red;")
|
111 |
-
col2.button("Gemini", key="reGemini", on_click=None, disabled=True)
|
112 |
-
elif st.session_state.selected_model == 'Gemini':
|
113 |
-
col1.button("Gemma", key="reGemma", on_click=None, disabled=True)
|
114 |
-
col2.button("Gemini", key="reGemini", on_click=None, disabled=True, style="background-color: red;")
|
115 |
|
116 |
user_prompt = st.chat_input("Pregunta acerca del contenido en el archivo PDF:")
|
117 |
if user_prompt:
|
|
|
91 |
st.success("Done")
|
92 |
|
93 |
# Model selection buttons
|
94 |
+
st.subheader("Select Model:")
|
95 |
+
model_selected = st.session_state.get('model_selected', '')
|
96 |
col1, col2 = st.columns(2)
|
97 |
+
if col1.button("GEMMA", key="gemmabtn"):
|
98 |
+
st.session_state.model_selected = "GEMMA"
|
99 |
+
if col2.button("GEMINI", key="geminibtn"):
|
100 |
+
st.session_state.model_selected = "GEMINI"
|
101 |
+
|
102 |
+
if model_selected:
|
103 |
+
st.write(f"Selected Model: **{model_selected}**")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
104 |
|
105 |
user_prompt = st.chat_input("Pregunta acerca del contenido en el archivo PDF:")
|
106 |
if user_prompt:
|