mychen76's picture
add usage
2912643
metadata
tags:
  - autotrain
  - text-generation
widget:
  - text: I love Train LLM because keep knowledge growth!

Medical Terminology Model

trained using dataset Credit :wiki_medical_terms

Model Usage:

from transformers import AutoTokenizer
import transformers
import torch

model = "mychen76/llama2_medical_terms"

tokenizer = AutoTokenizer.from_pretrained(model)
pipeline = transformers.pipeline(
    "text-generation",
    model=model,
    torch_dtype=torch.float16,
    device_map="auto",
)

sequences = pipeline(
    'Question:\nWhat is Occipital neuralgia and explain in detail?\nAnswer:\n',
    do_sample=True,
    top_k=10,
    num_return_sequences=1,
    eos_token_id=tokenizer.eos_token_id,
    max_length=200,
)
for seq in sequences:
    print(f"Result: {seq['generated_text']}")

Model Response:

Question: What is Occipital neuralgia and explain in detail? Answer: Occipital neuralgia is a type of headache disorder that is caused by pain in the occipital nerves, which are located in the back of the head and neck. These nerves transmit sensory information from the head and face to the brain. When the occipital nerves become irritated or compressed, it can cause intense pain in the back of the head, neck, and sometimes the eyes.

Causes: Occipital neuralgia can be caused by a variety of factors, including:

  1. compression of the occipital nerves due to a muscle spasm or inflammation in the neck
  2. injury to the occipital nerves, such as a blow to the head or a fall
  3. a narrowing of the spinal canal, known as spinal sten