Update app.py

app.py

@@ -73,7 +73,7 @@ def get_filename():
     choice = np.random.choice(embeddings.index)
     url = get_s3_url(choice)
     if check_image_url_accessible(url):
-        st.toast(url)
+        
         return 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 = {'num_leaves':10, 'objective': 'binary', 'metric' : 'binary'}
+    param = {'num_leaves':10, 'objective': 'binary', 'metric' : 'binary', 'lambda_l2': 0.1}
     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]