jianuo commited on
Commit
7253da8
·
1 Parent(s): 1287407

strong fix bugs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -367,7 +367,7 @@ def detect(input_text, args, tokenizer, return_green_token_mask=True):
367
  def run_gradio(args, model=None, tokenizer=None):
368
  """定义并启动gradio演示界面"""
369
  check_prompt_partial = partial(check_prompt, model=model)
370
- generate_partial = partial(generate, model=model)
371
  detect_partial = partial(detect)
372
 
373
  css = """
 
367
  def run_gradio(args, model=None, tokenizer=None):
368
  """定义并启动gradio演示界面"""
369
  check_prompt_partial = partial(check_prompt, model=model)
370
+ generate_partial = spaces.GPU(partial(generate, model=model))
371
  detect_partial = partial(detect)
372
 
373
  css = """