Spaces:
Running
Running
Commit
·
72f4061
1
Parent(s):
29212c7
Adding submission button and gemini model option
Browse files
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 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
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()
|