Update app.py

app.py

@@ -39,6 +39,7 @@ 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,6 +55,10 @@ st.divider()
 
 st.subheader("Classification")
 binary = st.toggle("binary classification")
+if binary:
+    classifictaion = "binary classification"
+else:
+    classification = "multi-class classification"
 st.divider()
 
 file_name = st.file_uploader("Choose an image...")