Manofem commited on
Commit
e23a911
·
verified ·
1 Parent(s): 615e4f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -7,8 +7,10 @@ llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardLM-7B-uncensored-GGUF
7
 
8
 
9
  def generate_response(message):
10
- response = llm(message)
11
- yield response
 
 
12
 
13
  def chatbot(message, history):
14
  response_generator = generate_response(message)
 
7
 
8
 
9
  def generate_response(message):
10
+
11
+ for text in llm(message):
12
+ yield text
13
+ yi
14
 
15
  def chatbot(message, history):
16
  response_generator = generate_response(message)