Update

app.py

@@ -14,7 +14,7 @@ import gradio_helpers
 import models
 import paligemma_parse
 
-INTRO_TEXT = """🤲 PaliGemma demo\n\n
+INTRO_TEXT = """🤲 PaliGemma GGUF demo\n\n
 | [Paper](https://arxiv.org/abs/2407.07726)
 | [GitHub](https://github.com/google-research/big_vision/blob/main/big_vision/configs/proj/paligemma/README.md) 
 | [HF blog post](https://huggingface.co/blog/paligemma) 
@@ -98,7 +98,7 @@ def compute(image, prompt, model_name, sampler):
 
 def warmup(model_name):
   image = PIL.Image.new('RGB', [1, 1])
-  _ = compute(image, '', model_name + "-text-model-q4_k_m.gguf", 'greedy')
+  _ = compute(image, '', model_name, 'greedy')
 
 
 def reset():

models.py

@@ -29,7 +29,7 @@ MODELS = {
 
 MODELS_INFO = {
     'paligemma-3b-mix-224': (
-        'JAX/FLAX PaliGemma 3B weights, finetuned with 224x224 input images and 256 token input/output '
+        'GGUF PaliGemma 3B weights quantized in Q4_K_M Format, finetuned with 224x224 input images and 256 token input/output '
         'text sequences on a mixture of downstream academic datasets. The models are available in float32, '
         'bfloat16 and float16 format for research purposes only.'
     ),