flake8

Helsinki-NLP · Apr 15, 2024 · 89186bb · 89186bb
1 parent 327b55e
commit 89186bb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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,