Chrisyichuan commited on
Commit
7c79929
·
1 Parent(s): e0deddf

add temp to user space

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -67,6 +67,7 @@ with st.sidebar:
67
  dataset_choice = st.selectbox("Dataset", available_datasets, index=0)
68
  model_choice = st.selectbox("Model", list(MODELS_CONFIG.keys()))
69
  steps_per_sample = st.slider("Max Steps", 3, 20, 10)
 
70
 
71
  # Load dataset with error handling
72
  data_paths = get_data_paths(dataset_choice)
@@ -105,7 +106,7 @@ if start_button:
105
  progress_bar = st.progress(0)
106
 
107
  with GeoBot(
108
- model=model_class, model_name=config["model_name"], headless=True
109
  ) as bot:
110
  for i, sample in enumerate(test_samples):
111
  st.divider()
 
67
  dataset_choice = st.selectbox("Dataset", available_datasets, index=0)
68
  model_choice = st.selectbox("Model", list(MODELS_CONFIG.keys()))
69
  steps_per_sample = st.slider("Max Steps", 3, 20, 10)
70
+ temperature = st.slider("Temperature", 0.0, 2.0, 0.0, 0.1, help="Controls randomness in AI responses. 0.0 = deterministic, higher = more creative")
71
 
72
  # Load dataset with error handling
73
  data_paths = get_data_paths(dataset_choice)
 
106
  progress_bar = st.progress(0)
107
 
108
  with GeoBot(
109
+ model=model_class, model_name=config["model_name"], headless=True, temperature=temperature
110
  ) as bot:
111
  for i, sample in enumerate(test_samples):
112
  st.divider()