Update app.py

app.py

@@ -28,8 +28,8 @@ load_dotenv()
 api_token = os.getenv("HF_TOKEN")
 huggingface_hub.login(token=api_token)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-tokenizer = AutoTokenizer.from_pretrained('google/gemma-2-9b')
-model = AutoModelForCausalLM.from_pretrained('google/gemma-2-9b').to(device)
+tokenizer = AutoTokenizer.from_pretrained('google/gemma-2-2b')
+model = AutoModelForCausalLM.from_pretrained('google/gemma-2-2b').to(device)
 
 #@spaces.GPU
 def read_pdf(file_path):