Spestly commited on
Commit
f5ad422
Β·
verified Β·
1 Parent(s): dc45496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -12,7 +12,7 @@ def initialize_model():
12
  if pipe is None:
13
  pipe = pipeline(
14
  "text-generation",
15
- model="apexion-ai/Orion-V1-4B",
16
  torch_dtype=torch.float16,
17
  device_map="auto"
18
  )
@@ -65,18 +65,18 @@ def generate_response(message, history, max_length=512, temperature=0.7, top_p=0
65
 
66
  # Create the Gradio interface
67
  def create_interface():
68
- with gr.Blocks(title="Orion-V1-4B Chat", theme=gr.themes.Soft()) as demo:
69
  gr.Markdown("""
70
- # πŸš€ Orion-V1-4B Chat
71
 
72
- Chat with the Orion-V1-4B model by Apexion AI. This is a 4B parameter language model optimized for conversation.
73
 
74
- **Model:** `apexion-ai/Orion-V1-4B`
75
  """)
76
 
77
  chatbot = gr.Chatbot(
78
  height=400,
79
- placeholder="Start chatting with Orion-V1-4B...",
80
  label="Chat"
81
  )
82
 
@@ -144,14 +144,14 @@ def create_interface():
144
  gr.Markdown("""
145
  ---
146
 
147
- ### About Orion-V1-4B
148
 
149
- Orion-V1-4B is a 4 billion parameter language model developed by Apexion AI.
150
  It's designed for efficient text generation and conversation.
151
 
152
  **Features:**
153
  - 4B parameters for efficient inference
154
- - Optimized for conversational AI
155
  - Supports various text generation tasks
156
 
157
  This Space uses ZeroGPU for efficient GPU allocation.
 
12
  if pipe is None:
13
  pipe = pipeline(
14
  "text-generation",
15
+ model="apexion-ai/Nous-V1-4B",
16
  torch_dtype=torch.float16,
17
  device_map="auto"
18
  )
 
65
 
66
  # Create the Gradio interface
67
  def create_interface():
68
+ with gr.Blocks(title="Nous-V1-4B Chat", theme=gr.themes.Soft()) as demo:
69
  gr.Markdown("""
70
+ # πŸš€ Nous-V1-4B Chat
71
 
72
+ Chat with the Nous-V1-4B model by Apexion AI. This is a 4B parameter language model optimized for conversation.
73
 
74
+ **Model:** `apexion-ai/Nous-V1-4B`
75
  """)
76
 
77
  chatbot = gr.Chatbot(
78
  height=400,
79
+ placeholder="Start chatting with Nous-V1-4B...",
80
  label="Chat"
81
  )
82
 
 
144
  gr.Markdown("""
145
  ---
146
 
147
+ ### About Nous-V1-4B
148
 
149
+ Nous-V1-4B is a 4 billion parameter language model developed by Apexion AI.
150
  It's designed for efficient text generation and conversation.
151
 
152
  **Features:**
153
  - 4B parameters for efficient inference
154
+ - Optimizsd for conversational AI
155
  - Supports various text generation tasks
156
 
157
  This Space uses ZeroGPU for efficient GPU allocation.