Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ import requests
|
|
11 |
from datetime import datetime
|
12 |
import google.generativeai as genai # Import Gemini API
|
13 |
from tensorflow.keras.models import load_model
|
14 |
-
from vit import
|
15 |
# Load the model
|
16 |
vit_classifier = load_model('models/vit_updated.weights.h5') # Replace with your model file path
|
17 |
# Configuring Google Gemini API
|
|
|
11 |
from datetime import datetime
|
12 |
import google.generativeai as genai # Import Gemini API
|
13 |
from tensorflow.keras.models import load_model
|
14 |
+
from vit import create_vit_classifier
|
15 |
# Load the model
|
16 |
vit_classifier = load_model('models/vit_updated.weights.h5') # Replace with your model file path
|
17 |
# Configuring Google Gemini API
|