Update README.md

README.md

@@ -28,7 +28,8 @@ This llama model was trained 2x faster with [Unsloth](https://github.com/unsloth
 # Inference Code
 
 ```python
-    prompt = """Below is given a Question and context to solve the question. Provide the answer to the question from the context.
+import torch
+prompt = """Below is given a Question and context to solve the question. Provide the answer to the question from the context.
 ### Question:
 {}
 
@@ -37,6 +38,7 @@ This llama model was trained 2x faster with [Unsloth](https://github.com/unsloth
 
 ### Answer:
 {}"""
+if True:
     from unsloth import FastLanguageModel
     model, tokenizer = FastLanguageModel.from_pretrained(
         model_name = "Ashed00/Hindi_tuned_Llama-3.2-1B", # YOUR MODEL YOU USED FOR TRAINING