Update app.py

app.py

@@ -9,7 +9,7 @@ def generate_viz(text, target_class):
     text_batch=[]
     text_batch.append(text)
     token_import, html_page = generate_visual(text_batch=text_batch, target_class=target_class)
-    return token_importance, html1
+    return token_import, html_page
 
 title = "Explain BERT 😄"