Update app.py

app.py

@@ -4,6 +4,7 @@ import streamlit as st
 import torch
 from PIL import Image
 
+@st.cache
 # default: Load the model on the available device(s)
 model = Qwen2VLForConditionalGeneration.from_pretrained(
     "Qwen/Qwen2-VL-7B-Instruct", torch_dtype="auto", device_map="auto"
@@ -25,6 +26,7 @@ processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
 # max_pixels = 1280*28*28
 # processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct", min_pixels=min_pixels, max_pixels=max_pixels)
 
+@st.cache_data
 # Streamlit app title
 st.title("OCR Image Text Extraction")