Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 import torch
 import numpy as np
 import spaces
+
 # Load the Pixtral model and processor
 model_id = "mrcuddle/lumimaid-v0.2-8b-pixtral"
 processor = AutoProcessor.from_pretrained(model_id)
@@ -25,8 +26,11 @@ def generate_text(input_text="", image=None):
     # Prepare inputs
     inputs = processor(text=input_text, images=image_pil, return_tensors="pt").to("cuda")
 
+    # Debug: Print the keys of the inputs dictionary
+    print("Processor output keys:", inputs.keys())
+
     # Check if image tokens are generated
-    if 'image_token' not in inputs or inputs['image_token'].numel() == 0:
+    if 'input_ids' not in inputs or inputs['input_ids'].numel() == 0:
         return "Error: No image tokens generated. Please check the image and try again."
 
     # Generate output