no message

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-from transformers import pipeline
+import pipeline
 from PIL import Image
 from datasets import load_dataset, Image, list_datasets
-from PIL import Image
+
 
 MODELS = [
             "google/vit-base-patch16-224", #Classifição geral
@@ -73,7 +73,8 @@ def main():
             st.write("# FLAG 3")
             
             #modle instance
-            classifier_pipeline = pipeline('image-classification', model="nateraw/vit-age-classifier", device=0)
+            classifier_pipeline = pipeline('image-classification', model=chosen_model_name, device=0)
+            #classifier_pipeline = pipeline('image-classification', model="nateraw/vit-age-classifier", device=0)
             st.write("# FLAG 4")
 
             #classification