Hugging Face's logo

Spaces:
hari7261
/
TechChatBot
Runtime error

App Files Community
hari7261 commited on
Commit
9e5e66c
·
verified ·
1 Parent(s): 7e7d8a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
7
  HF_TOKEN = os.getenv("HF_TOKEN")
8
 
9
 
10
- MODEL_NAME = "hari7261/TechChat"
11
 
12
  print("Loading tokenizer and model...")
13
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
 
7
  HF_TOKEN = os.getenv("HF_TOKEN")
8
 
9
 
10
+ MODEL_NAME = "mistralai/Mistral-7B-v0.1"
11
 
12
  print("Loading tokenizer and model...")
13
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)