tunght commited on
Commit
bf74f5b
·
1 Parent(s): cd136b9

support multi output

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -58,7 +58,8 @@ models = ["gpt-4-turbo",
58
  "gpt-3.5-turbo",
59
  "claude-3-sonnet-20240229",
60
  "claude-3-opus-20240229",
61
- "llama3-70b-8192"]
 
62
 
63
  openai.api_key = os.environ["OPENAI_API_KEY"]
64
 
@@ -220,6 +221,12 @@ def add_output_click(*struct_ref):
220
  return struct_ref
221
 
222
 
 
 
 
 
 
 
223
  with gr.Blocks() as demo:
224
  with gr.Row():
225
  with gr.Column():
@@ -227,9 +234,11 @@ with gr.Blocks() as demo:
227
  image = gr.Gallery(label="Images")
228
  garment_type = gr.Textbox(label="Garment Type", value="all", lines=1, interactive=True)
229
  # language = gr.Dropdown(languages, value="American English", interactive=True, label="Language")
230
- model = gr.Dropdown(models, value="gpt-4-turbo", interactive=True, label="Model")
231
- temperature = gr.Slider(minimum=0., maximum=1.0, value=0., interactive=True, label="Temperature")
232
- submit = gr.Button(value="Submit")
 
 
233
 
234
  with gr.Column():
235
  struct_ref = [val for i in range(10) for val in
@@ -244,6 +253,7 @@ with gr.Blocks() as demo:
244
  json_output = gr.JSON(label="JSON Output")
245
  submit.click(generate, inputs=[feature, image, garment_type, model, temperature, *struct_ref],
246
  outputs=[md_output, json_output])
 
247
 
248
 
249
 
 
58
  "gpt-3.5-turbo",
59
  "claude-3-sonnet-20240229",
60
  "claude-3-opus-20240229",
61
+ #"llama3-70b-8192",
62
+ ]
63
 
64
  openai.api_key = os.environ["OPENAI_API_KEY"]
65
 
 
221
  return struct_ref
222
 
223
 
224
+ def show_advanced(model, temperature):
225
+ model = gr.Dropdown(models, value="gpt-4-turbo", interactive=True, label="Model", visible=True)
226
+ temperature = gr.Slider(minimum=0., maximum=1.0, value=0., interactive=True, label="Temperature", visible=True)
227
+ return model, temperature
228
+
229
+
230
  with gr.Blocks() as demo:
231
  with gr.Row():
232
  with gr.Column():
 
234
  image = gr.Gallery(label="Images")
235
  garment_type = gr.Textbox(label="Garment Type", value="all", lines=1, interactive=True)
236
  # language = gr.Dropdown(languages, value="American English", interactive=True, label="Language")
237
+ model = gr.Dropdown(models, value="gpt-4-turbo", interactive=True, label="Model", visible=False)
238
+ temperature = gr.Slider(minimum=0., maximum=1.0, value=0., interactive=True, label="Temperature", visible=False)
239
+ with gr.Row():
240
+ submit = gr.Button(value="Submit")
241
+ advanced = gr.Button(value="Advanced")
242
 
243
  with gr.Column():
244
  struct_ref = [val for i in range(10) for val in
 
253
  json_output = gr.JSON(label="JSON Output")
254
  submit.click(generate, inputs=[feature, image, garment_type, model, temperature, *struct_ref],
255
  outputs=[md_output, json_output])
256
+ advanced.click(show_advanced, inputs=[], outputs=[model, temperature])
257
 
258
 
259