Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from huggingface_hub import snapshot_download,hf_hub_download
|
|
9 |
hf_hub_download(repo_id="huangyuyang/chatglm2-6b-int4.flm",local_dir="./", filename="chatglm2-6b-int4.flm")
|
10 |
|
11 |
st.set_page_config(
|
12 |
-
page_title="
|
13 |
page_icon=":robot:",
|
14 |
layout='wide'
|
15 |
)
|
@@ -18,7 +18,7 @@ st.set_page_config(
|
|
18 |
@st.cache_resource
|
19 |
def get_model():
|
20 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True)
|
21 |
-
model = llm.model(chatglm2-6b-int4.flm)
|
22 |
#model = model.eval()
|
23 |
return tokenizer, model
|
24 |
|
|
|
9 |
hf_hub_download(repo_id="huangyuyang/chatglm2-6b-int4.flm",local_dir="./", filename="chatglm2-6b-int4.flm")
|
10 |
|
11 |
st.set_page_config(
|
12 |
+
page_title="玉刚四号-演示",
|
13 |
page_icon=":robot:",
|
14 |
layout='wide'
|
15 |
)
|
|
|
18 |
@st.cache_resource
|
19 |
def get_model():
|
20 |
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True)
|
21 |
+
model = llm.model("chatglm2-6b-int4.flm")
|
22 |
#model = model.eval()
|
23 |
return tokenizer, model
|
24 |
|