Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -26,13 +26,12 @@ from transformers import DPTFeatureExtractor, DPTForDepthEstimation
|
|
26 |
from controlnet_aux import OpenposeDetector
|
27 |
from controlnet_aux.open_pose.body import Body
|
28 |
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
print("YoloWorld can not be load")
|
36 |
|
37 |
try:
|
38 |
from groundingdino.models import build_model
|
@@ -133,7 +132,7 @@ def sample_image(pipe,
|
|
133 |
generator=None,
|
134 |
concept_models=None,
|
135 |
num_inference_steps=50,
|
136 |
-
guidance_scale=
|
137 |
controller=None,
|
138 |
face_app=None,
|
139 |
image=None,
|
@@ -660,7 +659,7 @@ def main(device, segment_type):
|
|
660 |
|
661 |
def parse_args():
|
662 |
parser = argparse.ArgumentParser('', add_help=False)
|
663 |
-
parser.add_argument('--pretrained_model', default='
|
664 |
parser.add_argument('--controlnet_path', default='./checkpoint/InstantID/ControlNetModel', type=str)
|
665 |
parser.add_argument('--face_adapter_path', default='./checkpoint/InstantID/ip-adapter.bin', type=str)
|
666 |
parser.add_argument('--openpose_checkpoint', default='thibaud/controlnet-openpose-sdxl-1.0', type=str)
|
|
|
26 |
from controlnet_aux import OpenposeDetector
|
27 |
from controlnet_aux.open_pose.body import Body
|
28 |
|
29 |
+
|
30 |
+
from inference.models import YOLOWorld
|
31 |
+
from src.efficientvit.models.efficientvit.sam import EfficientViTSamPredictor
|
32 |
+
from src.efficientvit.sam_model_zoo import create_sam_model
|
33 |
+
import supervision as sv
|
34 |
+
|
|
|
35 |
|
36 |
try:
|
37 |
from groundingdino.models import build_model
|
|
|
132 |
generator=None,
|
133 |
concept_models=None,
|
134 |
num_inference_steps=50,
|
135 |
+
guidance_scale=3.0,
|
136 |
controller=None,
|
137 |
face_app=None,
|
138 |
image=None,
|
|
|
659 |
|
660 |
def parse_args():
|
661 |
parser = argparse.ArgumentParser('', add_help=False)
|
662 |
+
parser.add_argument('--pretrained_model', default='stablediffusionapi/realism-engine-sdxl-v30', type=str)
|
663 |
parser.add_argument('--controlnet_path', default='./checkpoint/InstantID/ControlNetModel', type=str)
|
664 |
parser.add_argument('--face_adapter_path', default='./checkpoint/InstantID/ip-adapter.bin', type=str)
|
665 |
parser.add_argument('--openpose_checkpoint', default='thibaud/controlnet-openpose-sdxl-1.0', type=str)
|