Update app.py

app.py

@@ -34,9 +34,9 @@ def is_offensive(text):
 @st.cache_resource
 def load_model():
     """Load the Hugging Face model and tokenizer."""
+    tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
+    model = AutoModelForCausalLM.from_pretrained("google/gemma-2b",device_map="auto")
     model_name = "mistralai/Mistral-7B-Instruct-v0.3"  # Updated model name
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
     hf_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if device.type == "cuda" else -1)
     return HuggingFacePipeline(pipeline=hf_pipeline)