zoya-hammad commited on
Commit
769695a
·
1 Parent(s): 584778f

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ folder = "my-knowledge-base/"
33
 
34
  load_dotenv()
35
  HF_TOKEN = os.getenv("HF_TOKEN")
36
- MODEL_NAME = "mistralai/Mistral-7B-Instruct"
37
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME , token=HF_TOKEN)
38
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto")
39
 
 
33
 
34
  load_dotenv()
35
  HF_TOKEN = os.getenv("HF_TOKEN")
36
+ MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.3"
37
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME , token=HF_TOKEN)
38
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto")
39