Update app.py

app.py

@@ -25,19 +25,21 @@ def enhance_prompt(prompt, style="photorealistic"):
     )
     return result
     
-def mistral_nemo_call(prompt, API_TOKEN, model="mistralai/Mistral-Nemo-Instruct-2407"):
+def mistral_nemo_call(prompt, API_TOKEN, model="mistralai/Mistral-Nemo-Instruct-2407", style="photo-realistic"):
     
     client = InferenceClient(api_key=API_TOKEN)
     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
-    """,
+    """
     
     response = ""
     for message in client.chat_completion(
         model=model,
-        messages=[{"role": "system", "content": system_prompt},{"role": "user", "content": prompt}],
+        messages=[{"role": "system", "content": system_prompt},
+                  {"role": "user", "content": prompt}
+                 ],
         max_tokens=500,
         stream=True,
     ):