syedfaisalabrar commited on
Commit
8d53540
·
verified ·
1 Parent(s): 662c0b7

Update backPrompt.py

Browse files
Files changed (1) hide show
  1. 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, max_num=12):
24
- images = image_file
25
  transform = build_transform(input_size=input_size)
26
- pixel_values = [transform(image) for image in images]
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