alphayomega commited on
Commit
abe3e7a
1 Parent(s): 32e9978

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -21
app.py CHANGED
@@ -64,34 +64,23 @@ if "messages" not in st.session_state:
64
  st.session_state.messages = []
65
 
66
  if "selected_model" not in st.session_state:
67
- st.session_state.selected_model = None
68
-
69
- # Detalles de los modelos
70
- models = {
71
- "llama-3.1-70b-versatile": {"name": "LLaMA3.1-70b", "tokens": 4096, "developer": "Meta"},
72
- "llama-3.1-8b-instant": {"name": "LLaMA3.1-8b", "tokens": 4096, "developer": "Meta"},
73
- "llama3-70b-8192": {"name": "Meta Llama 3 70B", "tokens": 4096, "developer": "Meta"},
74
- "llama3-8b-8192": {"name": "Meta Llama 3 8B", "tokens": 4096, "developer": "Meta"},
75
- "llama3-groq-70b-8192-tool-use-preview": {"name": "Llama 3 Groq 70B Tool Use (Preview)", "tokens": 4096, "developer": "Groq"},
76
- "gemma-7b-it": {"name": "Gemma-7b-it", "tokens": 4096, "developer": "Google"},
77
- "mixtral-8x7b-32768": {
78
- "name": "Mixtral-8x7b-Instruct-v0.1",
79
- "tokens": 32768,
80
- "developer": "Mistral",
81
- },
82
- }
83
 
84
  # Diseño para la selección de modelo y slider de tokens
85
  col1, col2 = st.columns([1, 3]) # Ajusta la proporción para hacer la primera columna más pequeña
86
 
87
  with col1:
88
- model_option = st.selectbox(
89
  "Choose a model:",
90
- options=list(models.keys()),
91
- format_func=lambda x: models[x]["name"],
92
- index=0, # Predeterminado al primer modelo en la lista
 
93
  )
94
- max_tokens_range = models[model_option]["tokens"]
95
  max_tokens = st.slider(
96
  "Max Tokens:",
97
  min_value=512,
 
64
  st.session_state.messages = []
65
 
66
  if "selected_model" not in st.session_state:
67
+ st.session_state.selected_model = "mixtral-8x7b-32768"
68
+
69
+ # Modelo disponible
70
+ model_option = "mixtral-8x7b-32768"
71
+ max_tokens_range = 32768
 
 
 
 
 
 
 
 
 
 
 
72
 
73
  # Diseño para la selección de modelo y slider de tokens
74
  col1, col2 = st.columns([1, 3]) # Ajusta la proporción para hacer la primera columna más pequeña
75
 
76
  with col1:
77
+ st.selectbox(
78
  "Choose a model:",
79
+ options=[model_option],
80
+ format_func=lambda x: "Mixtral-8x7b-Instruct-v0.1",
81
+ index=0, # Predeterminado al único modelo disponible
82
+ disabled=True # Desactivar la selección
83
  )
 
84
  max_tokens = st.slider(
85
  "Max Tokens:",
86
  min_value=512,