Update app.py
Browse files
app.py
CHANGED
@@ -8,9 +8,9 @@ model_cache = {}
|
|
8 |
|
9 |
# Available models
|
10 |
AVAILABLE_MODELS = {
|
11 |
-
"Nous-
|
12 |
-
"Nous-
|
13 |
-
"Nous-
|
14 |
}
|
15 |
|
16 |
@spaces.GPU
|
@@ -124,20 +124,20 @@ def generate_response(message, history, model_name, max_length=512, temperature=
|
|
124 |
def create_interface():
|
125 |
with gr.Blocks(title="Multi-Model Chat", theme=gr.themes.Soft()) as demo:
|
126 |
gr.Markdown("""
|
127 |
-
# π Nous-
|
128 |
|
129 |
-
Chat with the Nous-
|
130 |
|
131 |
**Available Models:**
|
132 |
-
- Nous-
|
133 |
-
- Nous-
|
134 |
-
- Nous-
|
135 |
""")
|
136 |
|
137 |
with gr.Row():
|
138 |
model_selector = gr.Dropdown(
|
139 |
choices=list(AVAILABLE_MODELS.keys()),
|
140 |
-
value="Nous-
|
141 |
label="Select Model",
|
142 |
info="Choose which model to use for generation"
|
143 |
)
|
@@ -221,13 +221,13 @@ def create_interface():
|
|
221 |
gr.Markdown("""
|
222 |
---
|
223 |
|
224 |
-
### About the Nous-
|
225 |
|
226 |
-
**Nous-
|
227 |
|
228 |
-
**Nous-
|
229 |
|
230 |
-
**Nous-
|
231 |
|
232 |
All models are designed for conversational AI and support various text generation tasks. The 8B model provides more sophisticated responses but requires more computational resources.
|
233 |
|
|
|
8 |
|
9 |
# Available models
|
10 |
AVAILABLE_MODELS = {
|
11 |
+
"Nous-1-4B": "apexion-ai/Nous-1-4B",
|
12 |
+
"Nous-1-8B": "apexion-ai/Nous-1-8B",
|
13 |
+
"Nous-1-2B": "apexion-ai/Nous-1-2B",
|
14 |
}
|
15 |
|
16 |
@spaces.GPU
|
|
|
124 |
def create_interface():
|
125 |
with gr.Blocks(title="Multi-Model Chat", theme=gr.themes.Soft()) as demo:
|
126 |
gr.Markdown("""
|
127 |
+
# π Nous-1 Model Chat Interface
|
128 |
|
129 |
+
Chat with the Nous-1 models by Apexion AI.
|
130 |
|
131 |
**Available Models:**
|
132 |
+
- Nous-1-4B (4 billion parameters)
|
133 |
+
- Nous-1-8B (8 billion parameters)
|
134 |
+
- Nous-1-2B (2 billion parameters)
|
135 |
""")
|
136 |
|
137 |
with gr.Row():
|
138 |
model_selector = gr.Dropdown(
|
139 |
choices=list(AVAILABLE_MODELS.keys()),
|
140 |
+
value="Nous-1-4B",
|
141 |
label="Select Model",
|
142 |
info="Choose which model to use for generation"
|
143 |
)
|
|
|
221 |
gr.Markdown("""
|
222 |
---
|
223 |
|
224 |
+
### About the Nous-1 Models
|
225 |
|
226 |
+
**Nous-1-2B**: 2 billion parameter model by Apexion AI, designed for fast and quick infrencing
|
227 |
|
228 |
+
**Nous-1-4B**: 4 billion parameter model by Apexion AI, optimisd for efficient conversation and text generation
|
229 |
|
230 |
+
**Nous-1-8B**: 8 billion parameter model by Apexion AI, offering enhanced capabilities and better performance for complex tasks
|
231 |
|
232 |
All models are designed for conversational AI and support various text generation tasks. The 8B model provides more sophisticated responses but requires more computational resources.
|
233 |
|