gastroenterology-gemma-300m-emb

pip install -U sentence-transformers

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("yasserrmd/gastroenterology-gemma-300m-emb")
# Run inference
queries = [
    "What are the hepatic manifestations and treatment options for individuals with Alagille syndrome (ALGS)?",
]
documents = [
    'The hepatic manifestations of ALGS typically include severe pruritus and disfiguring xanthomas. Treatment options for ALGS-related liver disease primarily involve supportive measures, such as choleretic agents (e.g., ursodeoxycholic acid), medications for itch relief (e.g., cholestyramine, rifampin, naltrexone), and surgical biliary diversion procedures (e.g., partial internal biliary diversion, ileal exclusion). The Kasai procedure, commonly used for biliary atresia, is not recommended for ALGS individuals as it may worsen the outcome. Liver transplantation is also an option and has shown an 80% 5-year survival rate, with most affected individuals experiencing catch-up growth.',
    'The Kimura-Takemoto classification is a system used to assess gastric mucosal atrophy. It identifies different atrophic patterns based on characteristic features observed during endoscopy. These patterns include C1 (atrophic mucosa in the antrum only), C2 (atrophic mucosa at the gastric angle or in the lower corpus), C3 (atrophic mucosa also found in the upper corpus), O1 (atrophic mucosa surrounds the gastric cardia, but the folds of the great curvature are maintained), O3 (entire gastric mucosa is atrophic, and the folds of the greater curvature disappear), and O2 (an intermediate condition between O1 and O3).',
    'Probiotics are living microorganisms that can have beneficial effects on the body. Studies have shown that probiotics can effectively attenuate alcoholic liver disease and improve liver function. They achieve these effects by maintaining the balance of intestinal microbiota, activating the endogenous microbial community, and regulating the immune system. Probiotics have been used in the treatment of alcoholic liver disease, with some strains, such as Lactobacillus rhamnosus GG (LGG), showing effectiveness in increasing the number of beneficial bacteria in the gut and altering serum levels of certain liver enzymes and bilirubin.',
]
query_embeddings = model.encode_query(queries)
document_embeddings = model.encode_document(documents)
print(query_embeddings.shape, document_embeddings.shape)
# [1, 768] [3, 768]

# Get the similarity scores for the embeddings
similarities = model.similarity(query_embeddings, document_embeddings)
print(similarities)
# tensor([[0.6757, 0.0341, 0.0612]])