Update app.py

app.py

@@ -7,10 +7,11 @@ from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTok
 print("="*150)
 print("MODEL LOADED")
 st.title("img_captioning_app")
+model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 #st.text("Build with Streamlit and OpenCV")
 if "photo" not in st.session_state:
 	st.session_state["photo"]="not done"
-
+feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 c2, c3 = st.columns([2,1])
 def change_photo_state():
 	st.session_state["photo"]="done"
@@ -23,9 +24,7 @@ def load_image(img):
 	return im
 activities = ["About"]
 choice = st.sidebar.selectbox("Select Activty",activities)
-model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 uploaded_photo = c2.file_uploader("Upload Image",type=['jpg','png','jpeg'], on_change=change_photo_state)
-feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 camera_photo = c2.camera_input("Take a photo", on_change=change_photo_state)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")