Spaces:
Sleeping
Sleeping
jianuo
commited on
Commit
·
7253da8
1
Parent(s):
1287407
strong fix bugs
Browse files
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 = """
|