Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from PIL import Image
 from vit_model_test import CustomModel  # Ensure you import the correct class
-from Vit_Traning.py import Custom_VIT_Model
+from Vit_Training import Custom_VIT_Model
 
 # Initialize the model
 model = Custom_VIT_Model()