Spaces:
Sleeping
Sleeping
no message
Browse files
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(
|
| 103 |
if chosen_model_name is not None:
|
| 104 |
-
COLS[0].write(
|
|
|
|
| 105 |
|
| 106 |
#Dataset
|
| 107 |
-
shosen_dataset_name = COLS[0].selectbox(
|
| 108 |
if shosen_dataset_name is not None:
|
| 109 |
-
COLS[0].write(
|
|
|
|
| 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(
|
|
|
|
| 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)
|