Spaces:
Sleeping
Sleeping
Hide DeepSeek-R1 8B as option
Browse files
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,
|
333 |
-
model_choice_step3 = st.selectbox("Select the model for Step 3", [HF_MODEL_MISTRAL, HF_MODEL_LLAMA,
|
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 |
|