Update app.py

app.py

@@ -1,4 +1,5 @@
 from transformers import AutoTokenizer
+import gradio as gr
 import transformers
 import torch