Update app.py

app.py

@@ -8,7 +8,9 @@ import jax.numpy as jnp
 import numpy as np
 
 import flax.linen as nn
-from inference import PaliGemmaModel as pali_gemma_model
+from inference import PaliGemmaModel
+
+pali_gemma_model = PaliGemmaModel()  # Create an instance of the model
 
 COLORS = ['#4285f4', '#db4437', '#f4b400', '#0f9d58', '#e48ef1']