Update app.py

app.py

@@ -5,6 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoMode
 from components import caption_chain, tag_chain
 from components import pexels, utils
 import os
+import gradio as gr
 
 model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")
 tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-gpt4-xl")