Update app.py

app.py

@@ -64,8 +64,8 @@ css = """
 def get_messages_formatter_type(model_name):
     if "Llama" in model_name:
         return MessagesFormatterType.LLAMA_3
-    elif "Mistral" in model_name:
-        return MessagesFormatterType.MISTRAL
+    elif "gemma" in model_name:
+        return MessagesFormatterType.GEMMA_2
     else:
         raise ValueError(f"Unsupported model: {model_name}")