Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,12 +4,12 @@ from threading import Thread
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
# App Configuration
|
| 7 |
-
st.set_page_config(page_title="
|
| 8 |
|
| 9 |
# 1. Load Qwen3 (Cached for efficiency)
|
| 10 |
@st.cache_resource
|
| 11 |
def load_qwen3():
|
| 12 |
-
model_id = "Qwen/Qwen3-1.7B
|
| 13 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 14 |
model = AutoModelForCausalLM.from_pretrained(
|
| 15 |
model_id,
|
|
@@ -21,7 +21,7 @@ def load_qwen3():
|
|
| 21 |
tokenizer, model = load_qwen3()
|
| 22 |
|
| 23 |
# 2. System Prompt Selection (Main UI, No Sidebar)
|
| 24 |
-
st.title("⚡
|
| 25 |
|
| 26 |
system_options = {
|
| 27 |
"General Assistant": "You are a helpful and concise assistant.",
|
|
|
|
| 4 |
import torch
|
| 5 |
|
| 6 |
# App Configuration
|
| 7 |
+
st.set_page_config(page_title="Klove AI ChatBox", page_icon="⚡")
|
| 8 |
|
| 9 |
# 1. Load Qwen3 (Cached for efficiency)
|
| 10 |
@st.cache_resource
|
| 11 |
def load_qwen3():
|
| 12 |
+
model_id = "Qwen/Qwen3-1.7B"
|
| 13 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 14 |
model = AutoModelForCausalLM.from_pretrained(
|
| 15 |
model_id,
|
|
|
|
| 21 |
tokenizer, model = load_qwen3()
|
| 22 |
|
| 23 |
# 2. System Prompt Selection (Main UI, No Sidebar)
|
| 24 |
+
st.title("⚡ Klove AI ChatBox")
|
| 25 |
|
| 26 |
system_options = {
|
| 27 |
"General Assistant": "You are a helpful and concise assistant.",
|