Update app.py

app.py

@@ -8,8 +8,8 @@ def generate_viz(text, target_class):
     print(f"Target Class: {target_class}")
     text_batch=[]
     text_batch.append(text)
-    token_import = generate_visual(text_batch=text_batch, target_class=target_class)
-    return token_import
+    token_import, html_page = generate_visual(text_batch=text_batch, target_class=target_class)
+    return token_import, html_page
 
 title = "Explain BERT 😄"
 
@@ -22,7 +22,7 @@ iface = gr.Interface(fn=generate_viz, inputs=[
             info="0 is negative, 1 is positive",
         ),
     ],
-    outputs=[ gr.Text(label="Token_Importance"),],
+    outputs=[ gr.Text(label="Token_Importance"), gr.HTML(label="visualize sentence"),],
     title=title,
     allow_flagging="never",
     cache_examples=True,