Update app.py

app.py

@@ -11,10 +11,9 @@ import requests
 from datetime import datetime
 import google.generativeai as genai  # Import Gemini API
 from tensorflow.keras.models import load_model
-from vit import vit_classifier
-
+from vit import create_vit_classfier
 # Load the model
-model = load_model('vit_updated.weights.h5')  # Replace with your model file path
+vit_classifier = load_model('models/vit_updated.weights.h5')  # Replace with your model file path
 # Configuring Google Gemini API
 GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)