marcofrodl commited on
Commit
368d204
·
verified ·
1 Parent(s): 0b48fe4

changed: Separation of model display name and model name; default model is now Mistral Large; fixed typo in system prompt

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -71,9 +71,10 @@ with gr.Blocks(title="Mistral Playground", css=css) as mistral_playground:
71
 
72
  with gr.Row(variant='panel'):
73
  api_key = gr.Textbox(type='password', placeholder='Your Mistral AI API key', lines=1, label="Mistral AI API Key")
74
- model = gr.Radio(label="Model",
75
- choices=["open-mistral-7b", "open-mixtral-8x7b", "mistral-small-latest", "mistral-medium-latest", "mistral-large-latest"],
76
- value="mistral-medium-latest",
 
77
  )
78
  with gr.Row(variant='panel'):
79
  temperature = gr.Slider(minimum=0.01, maximum=1.0, value=0.2, step=0.1, label="Temperature")
@@ -81,7 +82,7 @@ with gr.Blocks(title="Mistral Playground", css=css) as mistral_playground:
81
  max_tokens = gr.Slider(minimum=1, maximum=16000, step=500, label="Max Tokens", value=4000)
82
 
83
  with gr.Row(variant='panel'):
84
- system = gr.Textbox(lines=2, label="System Message", value="You are a helpfull AI assistant.")
85
 
86
  with gr.Row(variant='panel'):
87
  gr.ChatInterface(
 
71
 
72
  with gr.Row(variant='panel'):
73
  api_key = gr.Textbox(type='password', placeholder='Your Mistral AI API key', lines=1, label="Mistral AI API Key")
74
+ model = gr.Radio(
75
+ label="Model",
76
+ choices=[["Mistral-7B","open-mistral-7b"], ["Mixtral-8x7B","open-mixtral-8x7b"], ["Mistral Small","mistral-small-latest"], ["Mistral Medium","mistral-medium-latest"], ["Mistral Large","mistral-large-latest"]],
77
+ value="mistral-large-latest",
78
  )
79
  with gr.Row(variant='panel'):
80
  temperature = gr.Slider(minimum=0.01, maximum=1.0, value=0.2, step=0.1, label="Temperature")
 
82
  max_tokens = gr.Slider(minimum=1, maximum=16000, step=500, label="Max Tokens", value=4000)
83
 
84
  with gr.Row(variant='panel'):
85
+ system = gr.Textbox(lines=2, label="System Message", value="You are a helpful AI assistant")
86
 
87
  with gr.Row(variant='panel'):
88
  gr.ChatInterface(