Update app.py

app.py

@@ -172,7 +172,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
 
     original_prompt = prompt
     if enhance_prompt_option:
-        prompt = enhance_prompt_v2(prompt, style=enhance_prompt_style)
+        prompt = mistral_nemo_call(prompt, style=enhance_prompt_style)
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
     if use_mistral_nemo:
         prompt = mistral_nemo_call(prompt, API_TOKEN=API_TOKEN, style=nemo_enhance_prompt_style)