Update app.py

app.py

@@ -3,10 +3,7 @@ from PIL import Image
 from vit_model_test import CustomModel  
 from vit_Training import Custom_VIT_Model
 
-# Initialize the model
-model = CustomModel()
 
-model_training = Custom_VIT_Model()
 
 
 custom_css = """
@@ -32,6 +29,13 @@ custom_css = """
 </style>
 """
 
+
+# Initialize the model
+model = CustomModel()
+
+model_training = Custom_VIT_Model()
+
+
 # Store the last prediction result
 last_prediction = None