Minor fix

app.py

@@ -1,4 +1,3 @@
-import argparse
 import os
 import json
 import datetime
@@ -7,30 +6,17 @@ import torch
 from transformers import AutoTokenizer, AutoModel
 
 
-def get_args():
-    parser = argparse.ArgumentParser(description='ChatGLM Arguments')
-
-    parser.add_argument('--path', default='THUDM/chatglm-6b-int4', help='The path of ChatGLM model')
-
-    return parser.parse_args()
-
-
-args = get_args()
-
-if not os.path.isdir(args.path):
-    raise FileNotFoundError('Model not found')
-
 if torch.cuda.is_available():
     device = 'cuda'
 else:
     device = 'cpu'
 
-tokenizer = AutoTokenizer.from_pretrained(args.path, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained('THUDM/chatglm-6b-int4', trust_remote_code=True)
 
 if device == 'cuda':
-    model = AutoModel.from_pretrained(args.path, trust_remote_code=True).half().cuda()
+    model = AutoModel.from_pretrained('THUDM/chatglm-6b-int4', trust_remote_code=True).half().cuda()
 else:
-    model = AutoModel.from_pretrained(args.path, trust_remote_code=True).float()
+    model = AutoModel.from_pretrained('THUDM/chatglm-6b-int4', trust_remote_code=True).float()
 
 model = model.eval()