Spaces:

SUPERFIRE777
/

NLP

Runtime error

SUPERFIRE777 commited on Mar 26, 2023

Commit

ce5c55e

1 Parent(s): 79912d7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ def func(query):
         model = RWKV("https://huggingface.co/BlinkDL/rwkv-4-pile-3b/resolve/main/RWKV-4-Pile-3B-Instruct-test1-20230124.pth")
         def generate(query):
-            model.loadContext(newctx=text)
             output = model.forward(number=100)["output"]
             return output
     except Exception as e:

         model = RWKV("https://huggingface.co/BlinkDL/rwkv-4-pile-3b/resolve/main/RWKV-4-Pile-3B-Instruct-test1-20230124.pth")
         def generate(query):
+            model.loadContext(newctx=query)
             output = model.forward(number=100)["output"]
             return output
     except Exception as e: