Alic-Li commited on
Commit
fab2a53
·
verified ·
1 Parent(s): 050bf24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -118,6 +118,10 @@ def chat_with_bot(message, history, temperature, top_p):
118
  with gr.Blocks(title="MiniRWKV_7 34.2M 🪿 2vGPU Space") as demo:
119
  gr.Markdown("# MiniRWKV_7 34.2M 🪿 ")
120
  gr.Markdown("### Only 34.2M Params!!! Use 2V CPU Backend to run this model. ")
 
 
 
 
121
 
122
  with gr.Row():
123
  with gr.Column(scale=3):
 
118
  with gr.Blocks(title="MiniRWKV_7 34.2M 🪿 2vGPU Space") as demo:
119
  gr.Markdown("# MiniRWKV_7 34.2M 🪿 ")
120
  gr.Markdown("### Only 34.2M Params!!! Use 2V CPU Backend to run this model. ")
121
+ @demo.load()
122
+ def on_load():
123
+ global model_state
124
+ model_state = None
125
 
126
  with gr.Row():
127
  with gr.Column(scale=3):