Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("TuringsSolutions/LegalGemmaV1", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("TuringsSolutions/LegalGemmaV1", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("TuringsSolutions/Gemma2LegalEdition", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("TuringsSolutions/Gemma2LegalEdition", trust_remote_code=True)
 
 def predict(prompt, temperature, max_tokens):
     inputs = tokenizer(prompt, return_tensors="pt")