Update app.py
Browse files
app.py
CHANGED
@@ -34,8 +34,8 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
34 |
wav2vec2_processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
|
35 |
wav2vec2_model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
|
36 |
|
37 |
-
vit_model = ViTForImageClassification.from_pretrained('
|
38 |
-
vit_feature_extractor = ViTFeatureExtractor.from_pretrained('
|
39 |
|
40 |
|
41 |
# Function to handle text input
|
|
|
34 |
wav2vec2_processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
|
35 |
wav2vec2_model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
|
36 |
|
37 |
+
vit_model = ViTForImageClassification.from_pretrained('ohidaoui/monuments-morocco-v1')
|
38 |
+
vit_feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')
|
39 |
|
40 |
|
41 |
# Function to handle text input
|