Update app.py

app.py

@@ -162,9 +162,9 @@ tokenizer.pad_token_id = 0
 
 ####################################################
 #Datensets für Finetuning laden
-#dataset_neu = daten_laden("alexkueck/tis") 
+dataset_neu = daten_laden("alexkueck/tis") 
 #dataset_neu = daten_laden("EleutherAI/pile") 
-dataset_neu = daten_laden("yelp_review_full")
+#dataset_neu = daten_laden("yelp_review_full")
 
 #############################################
 #Vorbereiten für das Training der neuen Daten
@@ -294,6 +294,7 @@ trainer = SetFitTrainer(
     train_dataset=lm_datasets["train"],
     eval_dataset=lm_datasets["test"],
     loss_class=CosineSimilarityLoss,
+    column_mapping = ['id', 'text'],
     batch_size=16,
     num_iterations=20, # Number of text pairs to generate for contrastive learning
     num_epochs=1 # Number of epochs to use for contrastive learning