Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import torch
 from transformers import pipeline
 
 model_ckpt = "HenryAI/KerasBERTv1"
@@ -13,13 +14,13 @@ def  cloze_task(text):
 answer = []      
 description="Let's see if BERT has learnt how to write Keras!"
 title="Keras BERT v1",
-gradio_ui = gr.Interface(fn= cloze_task,
+interface = gr.Interface(fn= cloze_task,
     title= title, description = description,
     inputs=[gr.inputs.Textbox(lines=3)],
     outputs=[gr.outputs.Textbox(label="Answer"),],
     examples=[["from tensorflow.keras import [MASK]"],],
     enable_queue=True
 )
-gradio_ui.launch(debug=True)
+interface.launch(debug=True)