no message

app.py

@@ -11,8 +11,8 @@ MODELS = [
 ]
 DATASETS = [
             "",
-            "NunT/vit-base-patch16-224", #Classifição geral
-            "NunT/vit-age-classifier" #Classifição de idade
+            "Nunt/testedata",
+            "Nunt/backup_leonardo_2024-02-01" 
 ]
 MAX_N_LABELS = 5
 
@@ -48,7 +48,7 @@ def main():
     '''
         Dataset
     '''          
-    shosen_dataset_name =st.radio("Select the model to use",  MODELS)
+    shosen_dataset_name = st.radio("Select the model to use",  DATASETS)
     if shosen_dataset_name is not None:
         st.write("You selected", shosen_dataset_name)
         image_object = dataset['pasta'][0]["image"]
@@ -56,7 +56,7 @@ def main():
     '''
         click to classify
     '''    
-    if shosen_model is not None and shosen_dataset is not None:
+    if shosen_model_name is not None and shosen_dataset_name is not None:
         st.image(shosen_dataset[0], caption="Uploaded Image", use_column_width=True)
         if st.button("Classify images"):
             dataset = load_dataset("Nunt/testedata","testedata_readme")