Spaces:
Sleeping
Sleeping
NeonBohdan
commited on
Commit
•
648a832
1
Parent(s):
3c042e6
Added Gemini
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
from neon_llm_chatgpt.chatgpt import ChatGPT
|
3 |
from neon_llm_palm2.palm2 import Palm2
|
|
|
4 |
from neon_llm_claude.claude import Claude
|
5 |
from utils import convert_history, PersonaConverter
|
6 |
import os
|
@@ -35,6 +36,15 @@ config = {
|
|
35 |
}
|
36 |
palm2 = Palm2(config)
|
37 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
key_anthropic = os.environ['ANTHROPIC_API_KEY']
|
39 |
config = {
|
40 |
"key": key_anthropic,
|
@@ -48,7 +58,8 @@ claude = Claude(config)
|
|
48 |
|
49 |
model_choices = {
|
50 |
"openai": chatgpt,
|
51 |
-
"
|
|
|
52 |
"anthropic": claude,
|
53 |
}
|
54 |
model_choices_list = list(model_choices.keys())
|
|
|
1 |
import gradio as gr
|
2 |
from neon_llm_chatgpt.chatgpt import ChatGPT
|
3 |
from neon_llm_palm2.palm2 import Palm2
|
4 |
+
from neon_llm_gemini.gemini import Gemini
|
5 |
from neon_llm_claude.claude import Claude
|
6 |
from utils import convert_history, PersonaConverter
|
7 |
import os
|
|
|
36 |
}
|
37 |
palm2 = Palm2(config)
|
38 |
|
39 |
+
config = {
|
40 |
+
"model": "gemini-pro",
|
41 |
+
"key_path": key_path,
|
42 |
+
"role": role,
|
43 |
+
"context_depth": 3,
|
44 |
+
"max_tokens": 256,
|
45 |
+
}
|
46 |
+
gemini = Gemini(config)
|
47 |
+
|
48 |
key_anthropic = os.environ['ANTHROPIC_API_KEY']
|
49 |
config = {
|
50 |
"key": key_anthropic,
|
|
|
58 |
|
59 |
model_choices = {
|
60 |
"openai": chatgpt,
|
61 |
+
"palm2": palm2,
|
62 |
+
"gemini": gemini,
|
63 |
"anthropic": claude,
|
64 |
}
|
65 |
model_choices_list = list(model_choices.keys())
|