Kajise Org commited on
Commit
e7f4482
1 Parent(s): 96cf2e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def generate(instruction):
55
  base_prompt = ins.format(instruction)
56
  feeding_data = search_ddg("What is KOIT-FM?")
57
 
58
- response = llm(ins.format(base_prompt + "\n" + feeding_data))
59
  result = response['choices'][0]['text']
60
  return result
61
 
 
55
  base_prompt = ins.format(instruction)
56
  feeding_data = search_ddg("What is KOIT-FM?")
57
 
58
+ response = llm(ins.format(base_prompt + "\n" + feeding_data), stop=['### Instruction:', '### End'])
59
  result = response['choices'][0]['text']
60
  return result
61