sguertl commited on
Commit
6e63545
·
verified ·
1 Parent(s): 90685aa

Hide DeepSeek-R1 8B as option

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -329,8 +329,8 @@ if openai_api_key and hf_api_key:
329
  client_hf_deepseek = InferenceClient(model=HF_MODEL_DEEPSEEK, token=hf_api_key)
330
 
331
  model_choice_step1 = st.selectbox("Select the model for Step 1", ["gpt-4o", "gpt-4o-mini"])
332
- model_choice_step2 = st.selectbox("Select the model for Step 2", [HF_MODEL_MISTRAL, HF_MODEL_LLAMA, HF_MODEL_DEEPSEEK, "gpt-4o", "gpt-4o-mini"])
333
- model_choice_step3 = st.selectbox("Select the model for Step 3", [HF_MODEL_MISTRAL, HF_MODEL_LLAMA, HF_MODEL_DEEPSEEK, "gpt-4o", "gpt-4o-mini"])
334
 
335
  st.subheader("Step 1: PlantUML Code Extraction using GPT-4o or GPT-4o Mini")
336
 
 
329
  client_hf_deepseek = InferenceClient(model=HF_MODEL_DEEPSEEK, token=hf_api_key)
330
 
331
  model_choice_step1 = st.selectbox("Select the model for Step 1", ["gpt-4o", "gpt-4o-mini"])
332
+ model_choice_step2 = st.selectbox("Select the model for Step 2", [HF_MODEL_MISTRAL, HF_MODEL_LLAMA, "gpt-4o", "gpt-4o-mini"])
333
+ model_choice_step3 = st.selectbox("Select the model for Step 3", [HF_MODEL_MISTRAL, HF_MODEL_LLAMA, "gpt-4o", "gpt-4o-mini"])
334
 
335
  st.subheader("Step 1: PlantUML Code Extraction using GPT-4o or GPT-4o Mini")
336