bp
Browse files
app.py
CHANGED
|
@@ -20,10 +20,8 @@ outdir = "/tmp/rayst3r"
|
|
| 20 |
print("Loading DINOv2 model")
|
| 21 |
dino_model = torch.hub.load('facebookresearch/dinov2', "dinov2_vitl14_reg")
|
| 22 |
dino_model.eval()
|
| 23 |
-
#dino_model.to("cuda")
|
| 24 |
|
| 25 |
print("Loading MoGe model")
|
| 26 |
-
#device = torch.device("cuda")
|
| 27 |
# Load the model from huggingface hub (or load from local).
|
| 28 |
moge_model = MoGeModel.from_pretrained("Ruicheng/moge-vitl")
|
| 29 |
|
|
@@ -178,8 +176,6 @@ def mask_rembg(input_img):
|
|
| 178 |
|
| 179 |
@GPU
|
| 180 |
def process_image(input_img):
|
| 181 |
-
|
| 182 |
-
|
| 183 |
# resize the input image
|
| 184 |
rotated = False
|
| 185 |
#if input_img.shape[0] > input_img.shape[1]:
|
|
|
|
| 20 |
print("Loading DINOv2 model")
|
| 21 |
dino_model = torch.hub.load('facebookresearch/dinov2', "dinov2_vitl14_reg")
|
| 22 |
dino_model.eval()
|
|
|
|
| 23 |
|
| 24 |
print("Loading MoGe model")
|
|
|
|
| 25 |
# Load the model from huggingface hub (or load from local).
|
| 26 |
moge_model = MoGeModel.from_pretrained("Ruicheng/moge-vitl")
|
| 27 |
|
|
|
|
| 176 |
|
| 177 |
@GPU
|
| 178 |
def process_image(input_img):
|
|
|
|
|
|
|
| 179 |
# resize the input image
|
| 180 |
rotated = False
|
| 181 |
#if input_img.shape[0] > input_img.shape[1]:
|