Update app.py
Browse files
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/
|
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(["
|
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():
|