From 89186bb68665770f00eba33e039293500fe54537 Mon Sep 17 00:00:00 2001
From: Mickus Timothee <mickus@lx2-400-27713.ad.helsinki.fi>
Date: Tue, 26 Mar 2024 08:14:23 +0200
Subject: [PATCH] flake8

---
 tools/demo/mammoth_demo.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tools/demo/mammoth_demo.py b/tools/demo/mammoth_demo.py
index 549c76bb..50cc4fdf 100644
--- a/tools/demo/mammoth_demo.py
+++ b/tools/demo/mammoth_demo.py
@@ -72,8 +72,6 @@
 </style>
 """
 
-
-
 ARCHITECTURE_HTML_ABNEG = """
 <h3>Decoder</h3>
 <div class="arch">
@@ -131,6 +129,7 @@
 </style>
 """
 
+
 def render(template, model_task):
     task, lang = model_task.split('_')
     if task == 'translate' or task == 'train':
@@ -176,7 +175,7 @@ def __call__(self):
                 height=None,
             )
         with col2:
-            architecture_html = ARCHITECTURE_HTML_HYDRA if not 'train' in model.task else ARCHITECTURE_HTML_ABNEG
+            architecture_html = ARCHITECTURE_HTML_HYDRA if 'train' not in model.task else ARCHITECTURE_HTML_ABNEG
             st.markdown(
                 render(architecture_html, model.task),
                 unsafe_allow_html=True,