Nymbo commited on
Commit
52906e9
·
1 Parent(s): bf97a50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ token = os.getenv("HF_TOKEN")
8
  def generate_text(prompt, model_choice, max_tokens, *other_params):
9
  # Сопоставление имени модели с URL
10
  model_urls = {
11
- "GPT-3.5": "https://api-inference.huggingface.co/models/ai-forever/ruGPT-3.5-13B",
12
  "GPT-4": "https://api-inference.huggingface.co/models/ai-forever/ruGPT-4"
13
  }
14
  # Выбор URL в зависимости от выбранной модели
@@ -38,7 +38,7 @@ with gr.Blocks() as demo:
38
  with gr.Tab("Базовые настройки"):
39
  with gr.Row():
40
  prompt = gr.Textbox(label="Prompt", lines=3, placeholder="Введите текст...")
41
- model_choice = gr.Radio(["GPT-3.5", "GPT-4"], label="Выбор модели", value="GPT-3.5")
42
 
43
  with gr.Tab("Расширенные настройки"):
44
  with gr.Row():
 
8
  def generate_text(prompt, model_choice, max_tokens, *other_params):
9
  # Сопоставление имени модели с URL
10
  model_urls = {
11
+ "GPT-3.5": "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
12
  "GPT-4": "https://api-inference.huggingface.co/models/ai-forever/ruGPT-4"
13
  }
14
  # Выбор URL в зависимости от выбранной модели
 
38
  with gr.Tab("Базовые настройки"):
39
  with gr.Row():
40
  prompt = gr.Textbox(label="Prompt", lines=3, placeholder="Введите текст...")
41
+ model_choice = gr.Radio(["Mixtral-8x7B", "GPT-4"], label="Выбор модели", value="GPT-3.5")
42
 
43
  with gr.Tab("Расширенные настройки"):
44
  with gr.Row():