I need more energy drinks!

app.py

@@ -37,7 +37,7 @@ theme = gr.themes.Monochrome(
 
 def generate(instruction): 
     result = ""
-    for x in llm(ins.format(instruction), max_tokens=2048, end=['### Instruction:', '### End'], stream=True):
+    for x in llm(ins.format(instruction), max_tokens=2048, stop=['### Instruction:', '### End'], stream=True):
         result += x['choices'][0]['text']
         yield result