Update app.py

app.py

@@ -1,9 +1,9 @@
-from transformers import pipeline,LlamaForCausalLM,AutoTokenizer
+from transformers import pipeline,GemmaForCausalLM,AutoTokenizer
 import gradio as gr
 import spaces
 # ignore_mismatched_sizes=True
 tokenizer = AutoTokenizer.from_pretrained('models/google/gemma-2-9b')
-model = LlamaForCausalLM.from_pretrained('models/google/gemma-2-9b',ignore_mismatched_sizes=True)
+model = GemmaForCausalLM.from_pretrained('models/google/gemma-2-9b',ignore_mismatched_sizes=True)
 pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
 
 @spaces.GPU(duration=120)