BestWishYsh commited on
Commit
a0cde36
·
verified ·
1 Parent(s): 0fc1484

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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,