|
|
|
@ -76,7 +76,11 @@
|
|
|
|
|
"import gradio as gr\n", |
|
|
|
|
"from clip_interrogator import Config, Interrogator\n", |
|
|
|
|
"\n", |
|
|
|
|
"ci = Interrogator(Config())\n", |
|
|
|
|
"config = Config()\n", |
|
|
|
|
"config.blip_offload = True\n", |
|
|
|
|
"config.chunk_size = 1024\n", |
|
|
|
|
"config.flavor_intermediate_count = 1024\n", |
|
|
|
|
"ci = Interrogator(config)\n", |
|
|
|
|
"\n", |
|
|
|
|
"def inference(image, mode):\n", |
|
|
|
|
" image = image.convert('RGB')\n", |
|
|
|
|