Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def Decoder1_optical_instance(image_input):
|
|
64 |
img = transforms.Resize([1024, 1024])(image_input)
|
65 |
transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229,0.224,0.225])])
|
66 |
input_img = transform(img)
|
67 |
-
input_img =
|
68 |
image_embedding = model.image_encoder(input_img) # torch.Size([1, 256, 64, 64])
|
69 |
sparse_embeddings, dense_embeddings, scatter_embeddings = model.prompt_encoder(
|
70 |
points=None,
|
|
|
64 |
img = transforms.Resize([1024, 1024])(image_input)
|
65 |
transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229,0.224,0.225])])
|
66 |
input_img = transform(img)
|
67 |
+
input_img = input_img.unsqueeze(0)
|
68 |
image_embedding = model.image_encoder(input_img) # torch.Size([1, 256, 64, 64])
|
69 |
sparse_embeddings, dense_embeddings, scatter_embeddings = model.prompt_encoder(
|
70 |
points=None,
|