Update app.py

app.py

@@ -8,6 +8,7 @@ def load_models():
     feature_extractor = ViTFeatureExtractor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
     tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
     return model, feature_extractor, tokenizer
+model, feature_extractor, tokenizer = load_models()
 #pickle.load(open('energy_model.pkl', 'rb'))
 #vocab = np.load('w2i.p', allow_pickle=True)
 st.title("Image_Captioning_App")
@@ -24,9 +25,9 @@ def load_image(img):
 	return im
 uploaded_photo = c2.file_uploader("Upload Image",type=['jpg','png','jpeg'], on_change=change_photo_state)
 camera_photo = c2.camera_input("Take a photo", on_change=change_photo_state)
+
 #st.subheader("Detection")
 if st.checkbox("Generate_Caption"):
-   model, feature_extractor, tokenizer = load_models()
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    model.to(device)
    max_length = 16