Update app.py

app.py

@@ -12,7 +12,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 st.title('🦜Seon\'s Legal QA For Dummies 🔗 ')
 
 
-model = AutoModelForCausalLM.from_pretrained("PyaeSoneK/finetuned_pythia-2.8b-deduped_legal",
+model = AutoModelForCausalLM.from_pretrained("PyaeSoneK/Fine_Tuned_Pythia_smallest_140_legal",
                                     device_map='auto',
                                   torch_dtype=torch.float32,
                                   use_auth_token= st.secrets['hf_access_token'],
@@ -22,7 +22,7 @@ model = AutoModelForCausalLM.from_pretrained("PyaeSoneK/finetuned_pythia-2.8b-de
   
                                             #  load_in_4bit=True
 
-tokenizer = AutoTokenizer.from_pretrained("PyaeSoneK/finetuned_pythia-2.8b-deduped_legal",
+tokenizer = AutoTokenizer.from_pretrained("PyaeSoneK/Fine_Tuned_Pythia_smallest_140_legal",
                                           use_auth_token=st.secrets['hf_access_token'],)
 
 # Use a pipeline for later