Update app.py
Browse files
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.
|
|
|
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:
|