Update backPrompt.py

backPrompt.py

@@ -20,11 +20,9 @@ def build_transform(input_size):
     return transform
 
 
-def load_image(image_file, input_size=800, max_num=12):
-    images = image_file
+def load_image(image_file, input_size=800):
     transform = build_transform(input_size=input_size)
-    pixel_values = [transform(image) for image in images]
-    pixel_values = torch.stack(pixel_values)
+    pixel_values = transform(image_file).unsqueeze(0)  # Add batch dimension
     return pixel_values