done

app.py

@@ -36,27 +36,26 @@ def main():
     st.title("Bulk Image Classification")
     st.markdown("This app uses several 🤗 models to classify images stored in 🤗 datasets.")
     st.write("Soon we will have a dataset template")
-    st.write("**Soon we will have dataset selector**")   
     
 
     '''
         Model
     '''
     shosen_model_name = st.selectbox("Select the model to use",  MODELS)
-    if shosen_model is not None:
+    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",  MODELS)
-    if shosen_dataset is not None:
+    if shosen_dataset_name is not None:
         st.write("You selected", shosen_dataset_name)
         image_object = dataset['pasta'][0]["image"]
-   
-        
-        
-        
+           
+    '''
+        click to classify
+    '''    
     if shosen_model is not None and shosen_dataset is not None:
         st.image(shosen_dataset[0], caption="Uploaded Image", use_column_width=True)
         if st.button("Classify images"):