Update main.py

main.py

@@ -332,4 +332,4 @@ def get_text_embeddings(valid_df):
             text_embeddings = model.text_projection(text_features)
             valid_text_embeddings.append(text_embeddings)
 
-    return model, torch.cat(valid_text_embeddings)
+    return model, torch.cat(valid_text_embeddings)