Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import (
 )
 
 # Initialize model and tokenizer
-MODEL_NAME = "google/gemma-2b"  # You can change this to your preferred model
+MODEL_NAME = "google/gemma-2-2b"  # You can change this to your preferred model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)