acecalisto3 commited on
Commit
c870f5a
·
verified ·
1 Parent(s): 1dd5fcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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 = "OpenBMB/multilingual-codeparrot"
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
- "‎apple/OpenELM",
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
  ],