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()
|