Cran-May commited on
Commit
244e7af
·
1 Parent(s): 19e10b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="ChatGLM2-6b 演示",
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