no message

app.py

@@ -39,26 +39,27 @@ def main():
     '''
         Model
     '''
-    shosen_model_name = st.selectbox("Select the model to use",  MODELS)
+    shosen_model_name = st.selectbox("Select the model to use",  MODELS, index=0)
     if shosen_model_name is not None:
         st.write("You selected", shosen_model_name) 
         
     '''
         Dataset
     '''          
-    shosen_dataset_name = st.radio("Select the model to use",  DATASETS)
+    shosen_dataset_name = st.radio("Select the model to use",  DATASETS index=0)
     if shosen_dataset_name is not None:
         st.write("You selected", shosen_dataset_name)
-        image_object = dataset['pasta'][0]
+        
+        #image_object = dataset['pasta'][0]
            
     '''
         click to classify
     '''    
     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")
-            classifier = pipeline('image-classification', model=model_name, device=0)
+            dataset = load_dataset(shosen_dataset_name,"testedata_readme")
+            st.image(dataset[0], caption="Uploaded Image", use_column_width=True)
+            classifier = pipeline('image-classification', model=shosen_model_name, device=0)
             classify_images(classifier, dataset)