Update app_merged.py

app_merged.py

@@ -616,7 +616,7 @@ def infer(
 ):
     #logging.info(f"Input image shape: {image.shape}, dtype: {image.dtype}")
 
-    clear_memory()
+
     
     # Convert input to PIL if needed
     if isinstance(image, np.ndarray):
@@ -624,6 +624,8 @@ def infer(
             image = Image.fromarray(image, 'RGBA')
         else:  # RGB
             image = Image.fromarray(image, 'RGB')
+
+    clear_memory()
     
     #logging.info(f"Converted to PIL Image mode: {image.mode}")
     
@@ -1055,12 +1057,12 @@ def use_orientation(selected_image:gr.SelectData):
 def process_image(input_image, input_text):
     """Main processing function for the Gradio interface"""
 
-
-     clear_memory()
-
     if isinstance(input_image, Image.Image):
         input_image = np.array(input_image)
 
+    
+    clear_memory()
+
     # Initialize configs
     API_TOKEN = "9c8c865e10ec1821bea79d9fa9dc8720"
     SAM2_CHECKPOINT = "./checkpoints/sam2_hiera_large.pt"