update app

app.py

@@ -5,15 +5,20 @@ from tqdm import tqdm
 
 st.title("Coffe machine captioning app")
 
-with st.spinner('Loading model and tokenizer...'):
+@st.cache_resource()
+def load_model():
+    with st.spinner('Loading model and tokenizer...'):
 
-    model_id = "Fer14/paligemma_coffe_machine_caption"
+        model_id = "Fer14/paligemma_coffe_machine_caption"
 
-    model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
-    processor = PaliGemmaProcessor.from_pretrained(model_id)
+        model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
+        processor = PaliGemmaProcessor.from_pretrained(model_id)
 
-st.success('Model loaded!')
-   
+    st.success('Model loaded!')
+    return model, processor
+
+
+model, processor = load_model()
 
 
 # Instructions for Tesseract OCR