Update app.py

app.py

@@ -77,7 +77,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
         prompt = enhance_prompt(prompt)
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
     if use_mistral_nemo:
-        prompt = mistral_nemo_call(prompt,API_TOKEN=API_TOKEN)
+        prompt = mistral_nemo_call(prompt,API_TOKEN=API_TOKEN, model='meta-llama/Llama-3.2-1B',style="cartoon")
         print(f'\033[1mGeneration {key} Mistral-Nemo prompt:\033[0m {prompt}')
         
     final_prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
@@ -165,7 +165,7 @@ css1 = """
     max-width: 600px;
     margin-left: auto;
     margin-right: auto;
-    background-color: rgba(255, 255, 255, 0.8);  /* semi-transparent white */
+    background-color: rgba(255, 255, 255, 0.001);  /* semi-transparent white */
     padding: 20px;
     border-radius: 10px;
 }
@@ -184,7 +184,6 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
     gr.HTML("""
         <center>
             <div id="title-container">
-                <img id="title-icon" src="icon.jpg" alt="Icon">
                 <h1 id="title-text">FLUX Capacitor</h1>
             </div>
         </center>
@@ -216,6 +215,6 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
             final_prompt_output = gr.Textbox(label="Final Prompt", elem_id="final-prompt-output")
         
         # Adjust the click function to include the API key, use_dev, and enhance_prompt_option as inputs
-        text_button.click(query, inputs=[text_prompt, negative_prompt, steps, cfg, method, seed, strength, huggingface_api_key, use_dev, enhance_prompt_option], outputs=[image_output, seed_output, final_prompt_output])
+        text_button.click(query, inputs=[text_prompt, negative_prompt, steps, cfg, method, seed, strength, huggingface_api_key, use_dev, enhance_prompt_option, use_mistral_nemo], outputs=[image_output, seed_output, final_prompt_output])
 
 app.launch(show_api=True, share=False)