Create README.md

README.md

@@ -0,0 +1,30 @@
+
+## How to Load and Use the Model
+To use the model:
+
+1. Install required libraries: torch and transformers
+2. Use the following code:
+
+```python
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("Sourabh2/Chemistry_elements", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Sourabh2/Chemistry_elements", trust_remote_code=True)
+# Set up the device (GPU if available, otherwise CPU)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
+# Example usage
+messages = [
+    {
+        "role": "user",
+        "content": "hydrogen"
+    }
+]
+prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+inputs = tokenizer(prompt, return_tensors='pt', padding=True, truncation=True)
+outputs = model.generate(**inputs, max_length=150, num_return_sequences=1)
+text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+print(text.split("assistant")[1])
+# Decode and print output
+['Symbol: H', 'Atomic_Number: 1', 'Atomic_Weight: 1.008', 'Density: 0.0899', 'Melting_Point: 14.01', 'Boiling_Point: 20.28', 'Phase: Gas', 'Absolute_Melting_Point: 14.01']