Spaces:

Hazzzardous
/

RWKV-Instruct

Build error

Hazzzardous commited on Feb 7, 2023

Commit

7bc916c

1 Parent(s): 177a1d3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,8 +19,8 @@ def predict(input, history=None):
 def freegen(input):
     model.resetState()
-    return input,model.loadContext(newctx=input)["output"]
 with gr.Blocks() as demo:
     with gr.Tab("Chatbot"):
         chatbot = gr.Chatbot()

 def freegen(input):
     model.resetState()
+    model.loadContext(newctx=input)
+    return model.forward(number=100)["output"]
 with gr.Blocks() as demo:
     with gr.Tab("Chatbot"):
         chatbot = gr.Chatbot()