Update app.py

app.py

@@ -39,7 +39,6 @@ def get_prediction(img):
     return predictions[0,:]
 
 models = ['DDPM', 'Glide', 'Latent Diffusion', 'Palette', 'Stable Diffusion', 'VQ Diffusion', 'real', 'unseen_fake']
-classification = "multi-class classification"
 
 local_model_path = snapshot_download(repo_id="ElBeh/ma_basemodel")
 model = keras.models.load_model(local_model_path)
@@ -54,11 +53,7 @@ compress = st.toggle("jpeg compression")
 st.divider()
 
 st.subheader("Classification")
-binary = st.toggle(label=classification)
-if binary:
-    classification = "binary classification"
-else:
-    classification = "multi-class classification"
+binary = st.toggle("Activate binary classification")
 st.divider()
 
 file_name = st.file_uploader("Choose an image...")