Nuno-Tome commited on
Commit
2c39f53
Β·
1 Parent(s): 641cb95

removed restart button

Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -14,6 +14,8 @@ DATASETS = [
14
  ]
15
  MAX_N_LABELS = 5
16
  SPLIT_TO_CLASSIFY = 'pasta'
 
 
17
 
18
  #(image_object, classifier_pipeline)
19
  #def classify_one_image(classifier_model, dataset_to_classify):
@@ -49,7 +51,7 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
49
 
50
  #dataset
51
  dataset = load_dataset(shosen_dataset_name,"testedata_readme")
52
- with col2:
53
  #Image teste load
54
  image_object = dataset['pasta'][0]["image"]
55
  st.image(image_object, caption="Uploaded Image", width=300)
@@ -73,17 +75,16 @@ def classify_full_dataset(shosen_dataset_name, chosen_model_name):
73
 
74
  def main():
75
  st.title("Bulk Image Classification DEMO")
76
- col1, col2 = st.columns([3, 1])
77
 
78
- # Restart or reset your app
79
 
80
- if st.button("Restart"):
81
- # Code to restart or reset your app goes here
82
- import subprocess
83
- subprocess.call(["shutdown", "-r", "-t", "0"])
 
84
 
85
 
86
- with col1:
87
  st.markdown("This app uses several πŸ€— models to classify images stored in πŸ€— datasets.")
88
  st.write("Soon we will have a dataset template")
89
 
 
14
  ]
15
  MAX_N_LABELS = 5
16
  SPLIT_TO_CLASSIFY = 'pasta'
17
+ COL1, COL2 = st.columns([3, 1])
18
+
19
 
20
  #(image_object, classifier_pipeline)
21
  #def classify_one_image(classifier_model, dataset_to_classify):
 
51
 
52
  #dataset
53
  dataset = load_dataset(shosen_dataset_name,"testedata_readme")
54
+ with COL2:
55
  #Image teste load
56
  image_object = dataset['pasta'][0]["image"]
57
  st.image(image_object, caption="Uploaded Image", width=300)
 
75
 
76
  def main():
77
  st.title("Bulk Image Classification DEMO")
 
78
 
 
79
 
80
+ # Restart or reset your app
81
+ # if st.button("Restart"):
82
+ # # Code to restart or reset your app goes here
83
+ # import subprocess
84
+ # subprocess.call(["shutdown", "-r", "-t", "0"])
85
 
86
 
87
+ with COL1:
88
  st.markdown("This app uses several πŸ€— models to classify images stored in πŸ€— datasets.")
89
  st.write("Soon we will have a dataset template")
90