Update backPrompt.py
Browse files- backPrompt.py +2 -4
backPrompt.py
CHANGED
@@ -20,11 +20,9 @@ def build_transform(input_size):
|
|
20 |
return transform
|
21 |
|
22 |
|
23 |
-
def load_image(image_file, input_size=800
|
24 |
-
images = image_file
|
25 |
transform = build_transform(input_size=input_size)
|
26 |
-
pixel_values =
|
27 |
-
pixel_values = torch.stack(pixel_values)
|
28 |
return pixel_values
|
29 |
|
30 |
|
|
|
20 |
return transform
|
21 |
|
22 |
|
23 |
+
def load_image(image_file, input_size=800):
|
|
|
24 |
transform = build_transform(input_size=input_size)
|
25 |
+
pixel_values = transform(image_file).unsqueeze(0) # Add batch dimension
|
|
|
26 |
return pixel_values
|
27 |
|
28 |
|