Update app.py

app.py

@@ -1,6 +1,7 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import transformers
 import torch
+import gradio as gr
 
 # checkpoint = "tiiuae/falcon-40b-instruct"
 # checkpoint = "tiiuae/falcon-7b-instruct"