Update app.py

app.py

@@ -7,6 +7,13 @@ from transformers import AutoFeatureExtractor, AutoModel
 
 index = Index.from_env()
 
+
+model_ckpt = "google/vit-base-patch16-224-in21k"
+extractor = AutoFeatureExtractor.from_pretrained(model_ckpt)
+model = AutoModel.from_pretrained(model_ckpt)
+hidden_dim = model.config.hidden_size
+dataset = load_dataset("BounharAbdelaziz/Face-Aging-Dataset")
+
 # Data transformation chain.
 transformation_chain = T.Compose(
     [
@@ -17,12 +24,6 @@ transformation_chain = T.Compose(
     ]
 )
 
-model_ckpt = "google/vit-base-patch16-224-in21k"
-extractor = AutoFeatureExtractor.from_pretrained(model_ckpt)
-model = AutoModel.from_pretrained(model_ckpt)
-hidden_dim = model.config.hidden_size
-dataset = load_dataset("BounharAbdelaziz/Face-Aging-Dataset")
-
 with gr.Blocks() as demo:
     gr.Markdown(
         """