Update app.py

app.py

@@ -11,6 +11,7 @@ import transformers
 import torch
 import textwrap
 
+
 def main():
     st.set_page_config(page_title="Talk with BORROWER data")
     st.title("Talk with BORROWER data")
@@ -23,7 +24,7 @@ def main():
                     'delimiter': ','})
         data = loader.load()
 
-        model = "mistralai/Mistral-7B-v0.1"
+        model = "kittn/mistral-7B-v0.1-hf"
         tokenizer = AutoTokenizer.from_pretrained(model)
         pipeline = transformers.pipeline("text-generation", model=model, tokenizer=tokenizer, torch_dtype=torch.bfloat16, device_map="auto", do_sample=True, top_k=1, num_return_sequences=1, eos_token_id=tokenizer.eos_token_id,offload_folder="offload")