aadya1762 commited on
Commit
f318a82
·
1 Parent(s): c1e7456
Files changed (1) hide show
  1. app.py +0 -34
app.py CHANGED
@@ -22,10 +22,6 @@ def main():
22
  st.session_state.selected_model = "gemma-2b-it"
23
  if "selected_task" not in st.session_state:
24
  st.session_state.selected_task = "Question Answering"
25
- if "max_tokens" not in st.session_state:
26
- st.session_state.max_tokens = 512
27
- if "temperature" not in st.session_state:
28
- st.session_state.temperature = 0.7
29
 
30
  # Sidebar for login and configuration
31
  with st.sidebar:
@@ -57,36 +53,6 @@ def main():
57
  st.session_state.selected_task = selected_task
58
  st.rerun()
59
 
60
- # Model Config Selection
61
- new_max_tokens_value = st.slider(
62
- "Max Tokens",
63
- min_value=1,
64
- max_value=4096,
65
- value=st.session_state.max_tokens,
66
- step=1,
67
- )
68
- # After setting the slider values
69
- if st.session_state.max_tokens != new_max_tokens_value:
70
- st.session_state.max_tokens = new_max_tokens_value
71
- st.rerun()
72
-
73
- new_temperature_value = st.slider(
74
- "Temperature",
75
- min_value=0.0,
76
- max_value=1.0,
77
- value=st.session_state.temperature,
78
- step=0.01,
79
- )
80
- # After setting the slider values
81
- if st.session_state.temperature != new_temperature_value:
82
- st.session_state.temperature = new_temperature_value
83
- st.rerun()
84
-
85
- # Clear chat history button
86
- if st.button("Clear Chat History"):
87
- if "chat_instance" in st.session_state:
88
- st.session_state.chat_instance.clear_history()
89
- st.rerun()
90
 
91
  # Main content area
92
  # Initialize model with the selected configuration
 
22
  st.session_state.selected_model = "gemma-2b-it"
23
  if "selected_task" not in st.session_state:
24
  st.session_state.selected_task = "Question Answering"
 
 
 
 
25
 
26
  # Sidebar for login and configuration
27
  with st.sidebar:
 
53
  st.session_state.selected_task = selected_task
54
  st.rerun()
55
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
56
 
57
  # Main content area
58
  # Initialize model with the selected configuration