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