Update app.py

app.py

@@ -45,6 +45,29 @@ def enhance_prompt(prompt, model="mistralai/Mistral-7B-Instruct-v0.1", style="ph
     )
     return result
     """
+
+
+def enhance_prompt_v2(prompt, model="mistralai/Mistral-Nemo-Instruct-2407", style="photo-realistic"):
+    
+    client = Client("K00B404/Mistral-Nemo-custom")
+    
+    system_prompt=f"""
+    You are a image generation prompt enhancer specialized in the {style} style. 
+    You must respond only with the enhanced version of the users input prompt
+    Remember, image generation models can be stimulated by refering to camera 'effect' in the prompt like :4k ,award winning, super details, 35mm lens, hd
+    """
+    user_message=f"###input image generation prompt### {prompt}"
+   
+    result = client.predict(
+    		system_prompt=system_prompt,
+    		user_message=user_message,
+    		max_tokens=256,
+    		model_id=model,
+    		api_name="/predict"
+    )
+    return result
+
+    
 def mistral_nemo_call(prompt, API_TOKEN, model="mistralai/Mistral-Nemo-Instruct-2407", style="photo-realistic"):
     
     client = InferenceClient(api_key=API_TOKEN)
@@ -94,7 +117,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(prompt, style="cartoon")
+        prompt = enhance_prompt_v2(prompt, style="microscopic")
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
     if use_mistral_nemo:
         prompt = mistral_nemo_call(prompt,API_TOKEN=API_TOKEN,style="cartoon")