Update README.md

README.md

@@ -21,7 +21,7 @@ This model is for the module
 # Use a huggingafce pipeline as a high-level helper
 from transformers import pipeline
 import torch
-pipee = pipeline("text-generation",
+pipe = pipeline("text-generation",
                   model="SalehAhmad/Initial_Knowledge_Assessment_Test-Model-LLAMA7B_3Epochs",
                   device=torch.device('cuda' if torch.cuda.is_available() else 'cpu'),
                   torch_dtype=torch.bfloat16,
@@ -56,8 +56,9 @@ and so on.
 Paragraph: {paragraph}
  [/INST]</s>Response:'''
 
-output = pipee(prompt, 
+output = pipe(prompt, 
               num_return_sequences=1, 
               return_full_text=False)
+
 print(output[0]['generated_text'])
 ```