Update app.py
Browse files
app.py
CHANGED
@@ -83,8 +83,8 @@ if not isinstance(eva_transform_std, (list, tuple)):
|
|
83 |
eva_transform_mean = eva_transform_mean
|
84 |
eva_transform_std = eva_transform_std
|
85 |
|
86 |
-
face_main_model = FaceAnalysis(name='antelopev2', root=os.path.join(model_path, "face_encoder"), providers=['CUDAExecutionProvider'])
|
87 |
-
handler_ante = insightface.model_zoo.get_model(f'{model_path}/face_encoder/models/antelopev2/glintr100.onnx', providers=['CUDAExecutionProvider'])
|
88 |
face_main_model.prepare(ctx_id=0, det_size=(640, 640))
|
89 |
handler_ante.prepare(ctx_id=0)
|
90 |
|
@@ -227,7 +227,7 @@ with gr.Blocks() as demo:
|
|
227 |
with gr.Column():
|
228 |
with gr.Accordion("IPT2V: Face Input", open=True):
|
229 |
image_input = gr.Image(label="Input Image (should contain clear face)")
|
230 |
-
prompt = gr.Textbox(label="Prompt (Less than 200 Words)", placeholder="Enter your prompt here", lines=5)
|
231 |
with gr.Accordion("Examples", open=False):
|
232 |
examples_component_images = gr.Examples(
|
233 |
examples_images,
|
|
|
83 |
eva_transform_mean = eva_transform_mean
|
84 |
eva_transform_std = eva_transform_std
|
85 |
|
86 |
+
face_main_model = FaceAnalysis(name='antelopev2', root=os.path.join(model_path, "face_encoder"), providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
87 |
+
handler_ante = insightface.model_zoo.get_model(f'{model_path}/face_encoder/models/antelopev2/glintr100.onnx', providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
88 |
face_main_model.prepare(ctx_id=0, det_size=(640, 640))
|
89 |
handler_ante.prepare(ctx_id=0)
|
90 |
|
|
|
227 |
with gr.Column():
|
228 |
with gr.Accordion("IPT2V: Face Input", open=True):
|
229 |
image_input = gr.Image(label="Input Image (should contain clear face)")
|
230 |
+
prompt = gr.Textbox(label="Prompt (Less than 200 Words)", placeholder="Enter your prompt here. ConsisID has high requirements for prompt quality. You can use GPT-4o to refine the input text prompt, example can be found on our github.", lines=5)
|
231 |
with gr.Accordion("Examples", open=False):
|
232 |
examples_component_images = gr.Examples(
|
233 |
examples_images,
|