Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,13 @@ import os
|
|
5 |
import requests
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
|
|
|
|
|
|
|
|
|
|
8 |
# Load the pre-trained model and tokenizer
|
9 |
-
model_name = "
|
10 |
model = AutoModelForCodeGeneration.from_pretrained(model_name)
|
11 |
tokenizer = AutoTokenizerForCodeGeneration.from_pretrained(model_name)
|
12 |
|
@@ -173,7 +178,7 @@ with gr.Blocks() as demo:
|
|
173 |
|
174 |
model_dropdown = gr.Dropdown(
|
175 |
choices=[
|
176 |
-
"
|
177 |
"Gabriel/Swe-review-setfit-model",
|
178 |
"OpenBMB/multilingual-codeparrot"
|
179 |
],
|
|
|
5 |
import requests
|
6 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
7 |
|
8 |
+
)
|
9 |
+
client = InferenceClient(
|
10 |
+
"mistralai/Mixtral-8x7B-Instruct-v0.1"
|
11 |
+
)
|
12 |
+
|
13 |
# Load the pre-trained model and tokenizer
|
14 |
+
model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
15 |
model = AutoModelForCodeGeneration.from_pretrained(model_name)
|
16 |
tokenizer = AutoTokenizerForCodeGeneration.from_pretrained(model_name)
|
17 |
|
|
|
178 |
|
179 |
model_dropdown = gr.Dropdown(
|
180 |
choices=[
|
181 |
+
"mistralai/Mixtral-8x7B-Instruct-v0.1",
|
182 |
"Gabriel/Swe-review-setfit-model",
|
183 |
"OpenBMB/multilingual-codeparrot"
|
184 |
],
|