Update app.py

app.py

@@ -65,7 +65,7 @@ def get_filename():
         st.session_state.pred = st.session_state.preds[choice]
         url = get_s3_url(embeddings.index[choice])
         if check_image_url_accessible(url):
-            st.toast(url)
+            
             return embeddings.index[choice]
         else:
             return get_filename()
@@ -115,7 +115,7 @@ def train_model(X,labels):
         return
     train_data = lgb.Dataset(X, label=labels)
     num_round = 10
-    param = {'objective': 'binary', 'metric' : 'auc', 'lambda_l2': 0.5}
+    param = {'num_leaves': 30, 'objective': 'binary', 'metric' : 'binary'}
     bst = lgb.train(param, train_data, num_round)
     in_sample_preds = bst.predict(X)
     in_sample_score = np.corrcoef([in_sample_preds,np.array(labels)])[0][1]