Update README.md

README.md

@@ -24,6 +24,8 @@ Copy code
 from transformers import pipeline
 
 # Load the model from Hugging Face
+
+```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -34,19 +36,17 @@ model = AutoModelForCausalLM.from_pretrained(
     # llm_int8_enable_fp32_cpu_offload=True
 )
 tokenizer = AutoTokenizer.from_pretrained("pitangent-ds/academic_phy")
-
+```
 # Perform inference
+```python
 text = "What are units ?"
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model.generate(**inputs)
 decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 print(decoded_output)
+```
 
-# Ask a question
-question = "what are units?"
-response = qa_pipeline(data)
-print(response["answer"])
 
 # Training Details
 Dataset: Curated ICSE Physics content for Classes 9 and 10 textbooks