cigol123 commited on
Commit
202d927
·
verified ·
1 Parent(s): 5ee2d76

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from llama_cpp import Llama
3
+
4
+ llm = Llama(
5
+ model_path="mkllm-7b-instruct-q4_0.gguf",
6
+ n_ctx=2048
7
+ )
8
+
9
+ def chat(message, history):
10
+ response = llm.create_completion(
11
+ f"USER: {message}\nASSISTANT:",
12
+ max_tokens=512,
13
+ temperature=0.7
14
+ )
15
+ return response['choices'][0]['text']
16
+
17
+ demo = gr.ChatInterface(
18
+ chat,
19
+ title="MKLLM Chat",
20
+ )
21
+
22
+ demo.launch(server_name="0.0.0.0", server_port=7860)
23
+