Upload README.md

README.md

@@ -10,7 +10,7 @@ license_link: https://ai.google.dev/gemma/terms
 ---
 
 <p align="center">
-<img width="300px" alt="CodeGemma" src="https://huggingface.co/TechxGenus/CodeGemma-2b/resolve/main/CodeGemma.jpg">
+<img width="300px" alt="CodeGemma" src="https://huggingface.co/TechxGenus/CodeGemma-7b/resolve/main/CodeGemma.jpg">
 </p>
 
 ### CodeGemma
@@ -30,9 +30,9 @@ PROMPT = """### Instruction
 """
 instruction = <Your code instruction here>
 prompt = PROMPT.format(instruction=instruction)
-tokenizer = AutoTokenizer.from_pretrained("TechxGenus/CodeGemma-2b")
+tokenizer = AutoTokenizer.from_pretrained("TechxGenus/CodeGemma-7b")
 model = AutoModelForCausalLM.from_pretrained(
-    "TechxGenus/CodeGemma-2b",
+    "TechxGenus/CodeGemma-7b",
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
@@ -54,7 +54,7 @@ PROMPT = """<bos>### Instruction
 instruction = <Your code instruction here>
 prompt = PROMPT.format(instruction=instruction)
 generator = pipeline(
-    model="TechxGenus/CodeGemma-2b",
+    model="TechxGenus/CodeGemma-7b",
     task="text-generation",
     torch_dtype=torch.bfloat16,
     device_map="auto",