Update app.py

app.py

@@ -73,7 +73,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     if enhance_prompt_option:
         prompt = enhance_prompt(prompt)
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
-    elif use_mistral_nemo:
+    if use_mistral_nemo:
         prompt = mistral_nemo_call(prompt)
         print(f'\033[1mGeneration {key} Mistral-Nemo prompt:\033[0m {prompt}')
         
@@ -203,7 +203,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
                         huggingface_api_key = gr.Textbox(label="Hugging Face API Key (required for API calls)", placeholder="Enter your Hugging Face API Key here", type="password", elem_id="api-key")
                         use_dev = gr.Checkbox(label="Use Dev API", value=False, elem_id="use-dev-checkbox")
                         enhance_prompt_option = gr.Checkbox(label="Enhance Prompt", value=False, elem_id="enhance-prompt-checkbox")
-
+                        use_mistral_nemo = gr.Checkbox(label="Use Mistral Nemo", value=False, elem_id="use-mistral-checkbox")
         with gr.Row():
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
         with gr.Row():