p1atdev commited on
Commit
13f05ca
1 Parent(s): 7def78c

fix: continue button behavior

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -62,6 +62,8 @@ with gr.Blocks() as demo:
62
 
63
  ライトノベルの冒頭だけを学習した 400M パラメータの RetNet モデルのデモです。
64
 
 
 
65
  ### 参考:
66
 
67
  - https://github.com/syncdoth/RetNet
@@ -145,7 +147,7 @@ with gr.Blocks() as demo:
145
  continue_btn.click(
146
  fn=continue_generate,
147
  inputs=[
148
- input_text,
149
  max_tokens,
150
  do_sample,
151
  temperature,
 
62
 
63
  ライトノベルの冒頭だけを学習した 400M パラメータの RetNet モデルのデモです。
64
 
65
+ モデル: https://huggingface.co/p1atdev/LightNovel-Intro-RetNet-400M
66
+
67
  ### 参考:
68
 
69
  - https://github.com/syncdoth/RetNet
 
147
  continue_btn.click(
148
  fn=continue_generate,
149
  inputs=[
150
+ output_text,
151
  max_tokens,
152
  do_sample,
153
  temperature,