Update README.md

README.md

@@ -10,7 +10,7 @@ language:
 pipeline_tag: text-generation
 ---
 
-# Model Card for Model ID
+# Model Card for ReidLM
 
  ReidLM is a fine-tuned version of Meta's LLaMA 3 model, specifically optimized for generating high-quality, contextually accurate responses in the domain of rare diseases. 
  Utilizing the Evol-Instruct methodology, this model was fine-tuned with dataset of over 400 rare diseases.
@@ -19,6 +19,7 @@ pipeline_tag: text-generation
 
 ## Model Details
 
+
 ### Model Description
 
 - **Developed by:** MSRIT 
@@ -55,8 +56,26 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 ## How to Get Started with the Model
 
 Use the code below to get started with the model.
+## Use with Transformers AutoModelForCausalLM
+```
+import transformers
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# Load the pre-trained model and tokenizer
+model_name = "Tanvi03/ReidLM"  
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-Use with Transformers
+def generate_text(prompt, max_length=1000):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(inputs.input_ids, max_length=max_length, num_return_sequences=1)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+prompt = "Explain MEN-1 with respect to how it affects the pituitary gland. What is the other name for this syndrome?"
+generated_text = generate_text(prompt)
+print(generated_text)
+  ```
 
 
 
@@ -91,19 +110,19 @@ Use with Transformers
 
 <!-- This should link to a Dataset Card if possible. -->
 
-[More Information Needed]
+<!--[More Information Needed] --->
 
 <!---#### Factors
 
 <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
 
-[More Information Needed]
+<!--[More Information Needed]
 
 <!---#### Metrics
 
 <!-- These are the evaluation metrics being used, ideally with a description of why. -->
 
-[More Information Needed]
+<!--[More Information Needed]
 
 <!---### Results
 
@@ -117,7 +136,7 @@ Use with Transformers
 
 <!-- Relevant interpretability work for the model goes here -->
 
-[More Information Needed]
+<!--[More Information Needed]
 
 <!---## Environmental Impact