Add chatglm-6b model

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
+from transformers import AutoTokenizer, AutoModel
 
-def greet(name):
-    return "Hello " + name + "!!"
+def chat(text):
+    global model, tokenizer
+    response, history = model.chat(tokenizer, text, "", 64, 0.7, 0.95)
+    return response
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+
+
+
+if __name__ == '__main__':
+    tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
+    model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()
+    model.eval()
+    iface = gr.Interface(fn=chat, inputs="text", outputs="text")
+    iface.launch()