Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline, AutoTokenizer, GemmaForCasualLM
+from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
 import torch
 
 #Sidebar menu
@@ -26,7 +26,7 @@ if chatbot:
 
   
     tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
-    model = GemmaForCasualLM.from_pretrained(
+    model = AutoModelForQuestionAnswering.from_pretrained(
     "google/gemma-2-9b-it",
     device_map="auto",
     torch_dtype=torch.bfloat16)