Illia56 commited on
Commit
31b3603
1 Parent(s): 6d9ecc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -99,8 +99,8 @@ def sidebar():
99
  )
100
 
101
  # system_promptSide = st.text_input("Optional system prompt:")
102
- temperatureSide = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.9, step=0.05)
103
- max_new_tokensSide = st.slider("Max new tokens", min_value=0.0, max_value=4096.0, value=4096.0, step=64.0)
104
  # ToppSide = st.slider("Top-p (nucleus sampling)", min_value=0.0, max_value=1.0, value=0.6, step=0.05)
105
  # RepetitionpenaltySide = st.slider("Repetition penalty", min_value=0.0, max_value=2.0, value=1.2, step=0.05)
106
 
@@ -113,8 +113,8 @@ def predict(message: str) -> Any:
113
  response = client.predict(
114
  message,
115
  '',
116
- (temperatureSide or 0.9),
117
- (max_new_tokensSide or 4096),
118
  0.6,
119
  1.2,
120
  api_name="/chat_1"
 
99
  )
100
 
101
  # system_promptSide = st.text_input("Optional system prompt:")
102
+ # temperatureSide = st.slider("Temperature", min_value=0.0, max_value=1.0, value=0.9, step=0.05)
103
+ # max_new_tokensSide = st.slider("Max new tokens", min_value=0.0, max_value=4096.0, value=4096.0, step=64.0)
104
  # ToppSide = st.slider("Top-p (nucleus sampling)", min_value=0.0, max_value=1.0, value=0.6, step=0.05)
105
  # RepetitionpenaltySide = st.slider("Repetition penalty", min_value=0.0, max_value=2.0, value=1.2, step=0.05)
106
 
 
113
  response = client.predict(
114
  message,
115
  '',
116
+ 0.9,
117
+ 4096,
118
  0.6,
119
  1.2,
120
  api_name="/chat_1"