mcnaughtonadm commited on
Commit
72f4061
·
1 Parent(s): 29212c7

Adding submission button and gemini model option

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -7,6 +7,7 @@ from huggingface_hub import InferenceClient
7
  MODEL_API_KEY_MAP = {
8
  "gpt-3.5-turbo": "OPENAI_API_KEY",
9
  "gpt-4": "OPENAI_API_KEY",
 
10
  "claude-3-haiku-20240307": "ANTHROPIC_API_KEY",
11
  "claude-3-opus-20240229": "ANTHROPIC_API_KEY",
12
  "claude-3-sonnet-20240229": "ANTHROPIC_API_KEY",
@@ -38,6 +39,9 @@ with gr.Blocks() as iface:
38
  lines=2, placeholder="Enter your cheminformatics question..."
39
  )
40
 
 
 
 
41
  response_output = gr.Textbox(label="Agent Response")
42
 
43
  # Event Triggers
@@ -46,11 +50,11 @@ with gr.Blocks() as iface:
46
  model_name_dropdown,
47
  api_key_input,
48
  )
49
- api_key_input.change(
50
- agent_fn, [prompt_input, model_name_dropdown, api_key_input], response_output
51
- )
52
- prompt_input.change(
53
- agent_fn, [prompt_input, model_name_dropdown, api_key_input], response_output
54
- )
55
 
56
  iface.launch()
 
7
  MODEL_API_KEY_MAP = {
8
  "gpt-3.5-turbo": "OPENAI_API_KEY",
9
  "gpt-4": "OPENAI_API_KEY",
10
+ "gemini-pro": "GOOGLE_API_KEY",
11
  "claude-3-haiku-20240307": "ANTHROPIC_API_KEY",
12
  "claude-3-opus-20240229": "ANTHROPIC_API_KEY",
13
  "claude-3-sonnet-20240229": "ANTHROPIC_API_KEY",
 
39
  lines=2, placeholder="Enter your cheminformatics question..."
40
  )
41
 
42
+ with gr.Row():
43
+ submit_button = gr.Button("Submit")
44
+
45
  response_output = gr.Textbox(label="Agent Response")
46
 
47
  # Event Triggers
 
50
  model_name_dropdown,
51
  api_key_input,
52
  )
53
+
54
+ submit_button.click(
55
+ agent_fn,
56
+ [prompt_input, model_name_dropdown, api_key_input],
57
+ response_output,
58
+ )
59
 
60
  iface.launch()