katielink commited on
Commit
1005936
·
verified ·
1 Parent(s): d20efb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,8 @@ API_TOKEN = os.environ.get("API_TOKEN", None)
16
 
17
  model2endpoint = {
18
  "zephyr-7b-beta": "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
19
- "mistral-7b-v0.1": "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-v0.1"
 
20
  }
21
  model_names = list(model2endpoint.keys())
22
 
@@ -26,7 +27,6 @@ def randomize_seed_generator():
26
  return seed
27
 
28
 
29
-
30
  def get_total_inputs(inputs, chatbot, preprompt, user_name, assistant_name, sep):
31
  past = []
32
  for data in chatbot:
 
16
 
17
  model2endpoint = {
18
  "zephyr-7b-beta": "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
19
+ "mistral-7b-instruct-v0.2": "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
20
+ "mixtral-8x7b-instruct-v0.1", "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
21
  }
22
  model_names = list(model2endpoint.keys())
23
 
 
27
  return seed
28
 
29
 
 
30
  def get_total_inputs(inputs, chatbot, preprompt, user_name, assistant_name, sep):
31
  past = []
32
  for data in chatbot: