Nuno-Tome commited on
Commit
49572dd
·
1 Parent(s): 88c1db0

no message

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -99,14 +99,16 @@ def main():
99
  st.write("Soon we will have a dataset template")
100
 
101
  #Model
102
- chosen_model_name = COLS[0].selectbox(f"Select the model to use", MODELS, index=0)
103
  if chosen_model_name is not None:
104
- COLS[0].write(f"You selected", chosen_model_name)
 
105
 
106
  #Dataset
107
- shosen_dataset_name = COLS[0].selectbox(f"Select the dataset to use", DATASETS, index=0)
108
  if shosen_dataset_name is not None:
109
- COLS[0].write(f"You selected", shosen_dataset_name)
 
110
 
111
  #click to classify
112
  #image_object = dataset['pasta'][0]
@@ -115,7 +117,8 @@ def main():
115
 
116
  #classification_array =[]
117
  classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
118
- COLS[0].write(f"Classification result: {classification_result}")
 
119
  #classification_array.append(classification_result)
120
  #st.write("# FLAG 6")
121
  #st.write(classification_array)
 
99
  st.write("Soon we will have a dataset template")
100
 
101
  #Model
102
+ chosen_model_name = COLS[0].selectbox("Select the model to use", MODELS, index=0)
103
  if chosen_model_name is not None:
104
+ COLS[0].write("You selected")
105
+ COLS[0].write(chosen_model_name)
106
 
107
  #Dataset
108
+ shosen_dataset_name = COLS[0].selectbox("Select the dataset to use", DATASETS, index=0)
109
  if shosen_dataset_name is not None:
110
+ COLS[0].write("You selected")
111
+ COLS[0].write(shosen_dataset_name)
112
 
113
  #click to classify
114
  #image_object = dataset['pasta'][0]
 
117
 
118
  #classification_array =[]
119
  classification_result = classify_full_dataset(shosen_dataset_name, chosen_model_name)
120
+ COLS[0].write("Classification result {classification_result}")
121
+ COLS[0].write(classification_result)
122
  #classification_array.append(classification_result)
123
  #st.write("# FLAG 6")
124
  #st.write(classification_array)