Update app.py

app.py

@@ -13,14 +13,21 @@ def process_image(image, prompt):
     # Process the image and prompt using the processor
     inputs = processor(image.convert("RGB"), prompt, return_tensors="pt")
     
-    # Generate output from the model
-    output = model.generate(**inputs, max_new_tokens=20)
+    # Print the inputs to debug
+    print("Processor outputs:", inputs)
     
-    # Decode and return the output
-    decoded_output = processor.decode(output[0], skip_special_tokens=True)
-    
-    # Return the answer (exclude the prompt part from output)
-    return decoded_output[len(prompt):]
+    try:
+        # Generate output from the model
+        output = model.generate(**inputs, max_new_tokens=20)
+        
+        # Decode and return the output
+        decoded_output = processor.decode(output[0], skip_special_tokens=True)
+        
+        # Return the answer (exclude the prompt part from output)
+        return decoded_output[len(prompt):]
+    except IndexError as e:
+        print(f"IndexError: {e}")
+        return "An error occurred during processing."
 
 # Define the Gradio interface
 inputs = [