Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -337,23 +337,38 @@ with gr.Blocks(css=css) as demo:
|
|
| 337 |
generate_button.click(
|
| 338 |
generate_vanilla,
|
| 339 |
inputs=[prompt, num_inference_steps, guidance_scale],
|
| 340 |
-
outputs=[task_status]
|
|
|
|
| 341 |
)
|
| 342 |
|
| 343 |
generate_button_vs.click(
|
| 344 |
generate_vs,
|
| 345 |
inputs=[prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
|
| 346 |
-
outputs=[task_status]
|
|
|
|
| 347 |
)
|
| 348 |
|
| 349 |
-
enhance_button.click(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 350 |
|
| 351 |
-
refresh_button.click(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 352 |
demo.load(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], every=1)
|
| 353 |
|
| 354 |
-
refresh_tasks_button.click(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 355 |
demo.load(update_task_status, outputs=[task_status], every=5) # 每5秒自动刷新一次
|
| 356 |
|
| 357 |
if __name__ == "__main__":
|
| 358 |
-
demo.queue(max_size=10
|
| 359 |
-
demo.launch()
|
|
|
|
| 337 |
generate_button.click(
|
| 338 |
generate_vanilla,
|
| 339 |
inputs=[prompt, num_inference_steps, guidance_scale],
|
| 340 |
+
outputs=[task_status],
|
| 341 |
+
concurrency_limit=1
|
| 342 |
)
|
| 343 |
|
| 344 |
generate_button_vs.click(
|
| 345 |
generate_vs,
|
| 346 |
inputs=[prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
|
| 347 |
+
outputs=[task_status],
|
| 348 |
+
concurrency_limit=1
|
| 349 |
)
|
| 350 |
|
| 351 |
+
enhance_button.click(
|
| 352 |
+
enhance_prompt_func,
|
| 353 |
+
inputs=[prompt],
|
| 354 |
+
outputs=[prompt],
|
| 355 |
+
concurrency_limit=1
|
| 356 |
+
)
|
| 357 |
|
| 358 |
+
refresh_button.click(
|
| 359 |
+
update_server_status,
|
| 360 |
+
outputs=[cpu_status, memory_status, disk_status, gpu_status],
|
| 361 |
+
concurrency_limit=1
|
| 362 |
+
)
|
| 363 |
demo.load(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], every=1)
|
| 364 |
|
| 365 |
+
refresh_tasks_button.click(
|
| 366 |
+
update_task_status,
|
| 367 |
+
outputs=[task_status],
|
| 368 |
+
concurrency_limit=1
|
| 369 |
+
)
|
| 370 |
demo.load(update_task_status, outputs=[task_status], every=5) # 每5秒自动刷新一次
|
| 371 |
|
| 372 |
if __name__ == "__main__":
|
| 373 |
+
demo.queue(max_size=10)
|
| 374 |
+
demo.launch(max_threads=4) # 设置最大线程数为4,您可以根据需要调整这个值
|