Hazzzardous commited on
Commit
7bc916c
1 Parent(s): 177a1d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ def predict(input, history=None):
19
 
20
  def freegen(input):
21
  model.resetState()
22
- return input,model.loadContext(newctx=input)["output"]
23
-
24
  with gr.Blocks() as demo:
25
  with gr.Tab("Chatbot"):
26
  chatbot = gr.Chatbot()
 
19
 
20
  def freegen(input):
21
  model.resetState()
22
+ model.loadContext(newctx=input)
23
+ return model.forward(number=100)["output"]
24
  with gr.Blocks() as demo:
25
  with gr.Tab("Chatbot"):
26
  chatbot = gr.Chatbot()