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)
|