Upload models.py

models.py

@@ -11,10 +11,11 @@ import gradio_helpers
 import paligemma_bv
 
 
-ORGANIZATION = 'google'
+ORGANIZATION = 'Jegree'
 BASE_MODELS = [
-    ('paligemma-3b-mix-224-jax', 'paligemma-3b-mix-224'),
-    ('paligemma-3b-mix-448-jax', 'paligemma-3b-mix-448'),
+    # ('paligemma-3b-mix-224-jax', 'paligemma-3b-mix-224'),
+    # ('paligemma-3b-mix-448-jax', 'paligemma-3b-mix-448'),
+    ('myPaligem', 'fine-tuned-paligemma-3b-pt-224')
 ]
 MODELS = {
     # **{
@@ -29,21 +30,15 @@ MODELS = {
 }
 
 MODELS_INFO = {
-    'paligemma-3b-mix-224': (
+    'fine-tuned-paligemma-3b-pt-224': (
         'JAX/FLAX PaliGemma 3B weights, 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.'
     ),
-    'paligemma-3b-mix-448': (
-        'JAX/FLAX PaliGemma 3B weights, finetuned with 448x448 input images and 512 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.'
-    ),
 }
 
 MODELS_RES_SEQ = {
-    'paligemma-3b-mix-224': (224, 256),
-    'paligemma-3b-mix-448': (448, 512),
+    'fine-tuned-paligemma-3b-pt-224': (224, 128),
 }
 
 # "CPU basic" has 16G RAM, "T4 small" has 15 GB RAM.