made default model

app.py

@@ -17,6 +17,9 @@ from io import BytesIO
 model_name = 'output/training_OnlineConstrativeLoss-2023-03-14_01-24-44'
 model_name = 'output/training_OnlineConstrativeLoss-2023-03-17_23-15-52'
 model_name = 'output/training_OnlineConstrativeLoss-2023-03-17_23-50-15'
+model_name = 'output/training_OnlineConstrativeLoss-2023-03-25_23-48-35' # had 89% accur on eval set
+
+model_name = 'sentence-transformers/paraphrase-albert-base-v2'
 
 similarity_threshold = 0.9
 

train.py

@@ -29,7 +29,7 @@ model_name ='sentence-transformers/paraphrase-albert-base-v2'
 model = SentenceTransformer(model_name)
 num_epochs = 12
 # Smaller is generally better more accurate results.
-train_batch_size = 10
+train_batch_size = 8
 
 #As distance metric, we use cosine distance (cosine_distance = 1-cosine_similarity)
 distance_metric = losses.SiameseDistanceMetric.COSINE_DISTANCE