Ctaake commited on
Commit
3e59d25
·
verified ·
1 Parent(s): a90d3c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,11 +9,11 @@ from datetime import datetime
9
  # Model which is used
10
  checkpoint = "CohereForAI/c4ai-command-r-v01"
11
  checkpoint = "mistral-community/Mixtral-8x22B-v0.1"
12
- path_to_log = "log.txt"
 
13
  # Inference client with the model (And HF-token if needed)
14
  client = InferenceClient(checkpoint)
15
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
16
-
17
  if checkpoint == "mistral-community/Mixtral-8x22B-v0.1":
18
  # Tokenizer chat template correction(Only works for mistral models)
19
  chat_template = open("mistral-instruct.jinja").read()
 
9
  # Model which is used
10
  checkpoint = "CohereForAI/c4ai-command-r-v01"
11
  checkpoint = "mistral-community/Mixtral-8x22B-v0.1"
12
+ path_to_log = "FlaggedFalse.txt"
13
+
14
  # Inference client with the model (And HF-token if needed)
15
  client = InferenceClient(checkpoint)
16
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 
17
  if checkpoint == "mistral-community/Mixtral-8x22B-v0.1":
18
  # Tokenizer chat template correction(Only works for mistral models)
19
  chat_template = open("mistral-instruct.jinja").read()