Updated mistral model

app.py

@@ -9,8 +9,8 @@ st.markdown("Let's create a magical story just for you!")
 # Initialize the model
 @st.cache_resource
 def load_model():
-    model = AutoModelForCausalLM.from_pretrained("ajibawa-2023/Young-Children-Storyteller-Mistral-7B", torch_dtype=torch.float16)
-    tokenizer = AutoTokenizer.from_pretrained("ajibawa-2023/Young-Children-Storyteller-Mistral-7B")
+    model = AutoModelForCausalLM.from_pretrained("blockblockblock/Young-Children-Storyteller-Mistral-7B-bpw6", torch_dtype=torch.float16)
+    tokenizer = AutoTokenizer.from_pretrained("blockblockblock/Young-Children-Storyteller-Mistral-7B-bpw6")
     return model, tokenizer
 
 model, tokenizer = load_model()