Update frontPrompt.py

frontPrompt.py

@@ -20,9 +20,23 @@ def build_transform(input_size):
     return transform
 
 
+def preprocess_for_model(image):
+    """Prepares an image for the OpenGVLab model."""
+    
+    # Define the necessary image transformations
+    transform = T.Compose([
+        
+        T.ToTensor(),  # Convert to PyTorch Tensor
+        T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # Normalize
+    ])
+    
+    image = transform(image).unsqueeze(0)  # Add batch dimension
+    return image
+
+
 def load_image(image_file):
     transform = build_transform(input_size=800)
-    pixel_values = transform(image_file).unsqueeze(0)  # Add batch dimension
+    pixel_values = preprocess_for_model(image_file)
     return pixel_values