Update README.md

README.md

@@ -44,8 +44,14 @@ Gemma-2B Fine-Tuned Python Model is a deep learning model based on the Gemma-2B
 
 ## Inference
 ```python
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("suriya7/Gemma-2B-Finetuned-Python-Model")
+model = AutoModelForCausalLM.from_pretrained("suriya7/Gemma-2B-Finetuned-Python-Model")
+
 query = input('enter a query:')
-prompt_template = """
+prompt_template = f"""
 <start_of_turn>user based on given instruction create a solution\n\nhere are the instruction {query}
 <end_of_turn>\n<start_of_turn>model
 """
@@ -61,3 +67,4 @@ for i in tokenizer.decode(generated_ids[0], skip_special_tokens=True).split('<en
     ans+=i 
 cleaned_output = output.replace('<start_of_turn>', '')
 print(cleaned_output)
+```