Update app.py

app.py

@@ -6,6 +6,7 @@ import keras
 import keras_nlp
 import json
 import ipywidgets as widgets
+from IPython.display import display
 
 
 css = """