sentence-transformers How to use crazyjeannot/en-literary-gte-base with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("crazyjeannot/en-literary-gte-base", trust_remote_code=True)
sentences = [
"The family’s?’ The boy looked sceptically at him. ‘The one in front,’ he said, ‘has a bumper sticker about Jews not giving up any of Israel. You called him a Jew.’ ‘And so he is,’ Mengele said. ‘At least he looks like one.’ He smiled. ‘This is hardly the time to talk about what words I used. Go get the water, please, and I’ll call the police.’ The boy cleared his throat. ‘Would you sit down again?’ he said. ‘I’ll call them.’ ‘Bobby dear—’ ‘Pickles,’ the boy said; the Dobermanns rushed snarling at Mengele. He backed down onto the settee, forearms crossed before his face. ‘Ketchup!’ he cried.",
"While the raw data of the research is not bad, the interpretation is rendered moot. See Barbara Schneider and David Stevenson, The Ambitious Generation: America's Teenagers Motivated but Directionless (New Haven, Conn.: Yale University Press, 1999). 29Several book have been written in this vein. While the two in the main text are the most prominent, the following offer valuable insight and perspective as well: Adrian Nicole LeBlanc, Random Family: Love, Drugs, Trouble and Coming ofAge in the Bronx (New York, Scribner, 2003); Joanna Lipper, Growing Up Fast (New York: Picador, 2003); Neil Postman, The Disappearance of Childhood (New York: Vintage Books, 1994). 30Elinor Burkett, Another Planet.- A Year in the Life of a Suburban High School (New York: HarperCollins, 2001). 31Patricia Hersch, A Tribe Apart.: AJourney into the Heart ofAmerican Adolescence (New York: Ballantine, 1999), p. viii. The strength of this book is its in-depth walk with adolescents. Its weakness is in being a very narrow scope with limited application. Still, this qualitative work offers more depth than surveys.",
"This wasn’t his way. Normally, he was in, out and done. One day, one interrogation, one paycheck, and then shipped off to the next location. It was the way he liked it. Wham, bam, thank you, ma’am. As it was, they had already spent more than their fair share of time together. He’d hate to think the effect she’d have on him in a week if he did decide to take her to Montana. “Why don’t you try to call him again?” he asked, feeling sheepish in feigning such deplorable ignorance. She couldn’t find out what he knew or his role in this. She would hate him forever if she did.",
"He ran toward second. Felix chased after the ball. “Go! Go!” Ms. Benson called again. Eric ran from second to third. Felix grabbed the ball. Eric touched third base and ran toward home. Felix threw the ball to Sarah who was standing in front of home plate. The ball reached Sarah before Eric did."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
