Update README.md

README.md

@@ -337,22 +337,37 @@ Then you can load this model and run inference.
 ```python
 from sentence_transformers import SentenceTransformer
 
+def embed_query(texts:list, model):
+    return model.encode(sentences=texts, 
+                prompt="Instruct: Retrieve semantically similar text.\nQuery: ",
+                batch_size=32,
+                show_progress_bar=True,
+                normalize_embeddings=False,
+                convert_to_numpy=True,
+                device="cuda")
+def embed_documents(texts:list, model):
+    return model.encode(sentences=texts, 
+                batch_size=1024,
+                show_progress_bar=True,
+                normalize_embeddings=False,
+                convert_to_numpy=True,
+                device="cuda")
+
 # Download from the 🤗 Hub
 model = SentenceTransformer("Sami92/multiling-e5-large-instruct-claim-matching")
 # Run inference
-sentences = [
-    'Instruct: Retrieve semantically similar text.\nQuery: Die Pentagon-Mitarbeiter wechseln zum Überstundenmodus Internetnutzer stellten fest, dass es in der Gegend, in der sich die Abteilung befindet, einen starken Anstieg der Bestellungen für Pizzalieferungen gab und gleichzeitig die örtlichen Bars leer waren. Zuletzt geschah dies im April, kurz vor und nach dem iranischen Angriff auf Israel. Dann bemerkte The Sun die hohe Auslastung der Pizzerien von Papa Johns mit Bestellungen vom Pentagon. Unsere Website <URL> Unser Kanal auf Englisch <URL>',
+
+queries = ['Die Pentagon-Mitarbeiter wechseln zum Überstundenmodus Internetnutzer stellten fest, dass es in der Gegend, in der sich die Abteilung befindet, einen starken Anstieg der Bestellungen für Pizzalieferungen gab und gleichzeitig die örtlichen Bars leer waren. Zuletzt geschah dies im April, kurz vor und nach dem iranischen Angriff auf Israel. Dann bemerkte The Sun die hohe Auslastung der Pizzerien von Papa Johns mit Bestellungen vom Pentagon. Unsere Website <URL> Unser Kanal auf Englisch <URL>',]
+documents = [
     'Erinnern Sie sich an meinen Beitrag von vorhin über die massiven Cyberausfälle. Verbinden Sie das alles mit dem, was wir seit „Shot Heard Round the World“ am Samstag gesagt haben. BQQM BQQM BQQM BQQM 1776 1776 BRICS',
     'Heute sind offensichtlich ältere Flugzeuge am Himmel die entweder Turbinen Probleme haben oder schon elektrisch Fliegen... Die ganzen Lügen sind sooo offensichtlich das es schon weh tut sie bewusst zu ignorieren bzw nicht zu erkennen... Ihr wollt die Wahrheit <URL>',
 ]
-embeddings = model.encode(sentences)
-print(embeddings.shape)
-# [3, 1024]
+query_embeddings = embed_documents(texts=queries, model=model)
+doc_embeddings = embed_query(texts=documents, model=model)
 
 # Get the similarity scores for the embeddings
-similarities = model.similarity(embeddings, embeddings)
-print(similarities.shape)
-# [3, 3]
+similarities = model.similarity(query_embeddings, doc_embeddings)
+
 ```
 
 <!--