fixex

app.py

@@ -61,11 +61,13 @@ class CustomModelChat:
             )
 
             # Handle image input if provided
-            image_inputs = []
             if image is not None:
                 if image.mode != 'RGB':
                     image = image.convert('RGB')
-                image_inputs.append([image])
+                # Create a list of None for previous turns and add the current image
+                image_inputs = [None] * (len(messages) - 1) + [image]
+            else:
+                image_inputs = None
 
             # Prepare model inputs
             model_inputs = self.processor(