switch to 2b model

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import os
 from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
 token = os.environ["HF_TOKEN"]
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b",token=token)
-model = AutoModelForCausalLM.from_pretrained("google/gemma-7b",token=token)
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b",token=token)
+model = AutoModelForCausalLM.from_pretrained("google/gemma-2b",token=token)
 streamer = TextStreamer(tokenizer,skip_prompt=True)