katielink commited on
Commit
75612e3
1 Parent(s): c5daedb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ API_TOKEN = os.environ.get("API_TOKEN", None)
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
- "gemma-7b-it", "https://api-inference.huggingface.co/models/google/gemma-7b-it"
22
  }
23
  model_names = list(model2endpoint.keys())
24
 
 
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
+ "gemma-7b-it": "https://api-inference.huggingface.co/models/google/gemma-7b-it"
22
  }
23
  model_names = list(model2endpoint.keys())
24