Update app.py

app.py

@@ -11,8 +11,6 @@ def init_qwen_model():
     _processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
     return _model, _processor
 
-# Modified function to use only the image as the argument
-@st.cache_data
 def get_qwen_text(uploaded_file, model, processor):
     if uploaded_file is not None:
         # Open the uploaded image file
@@ -55,7 +53,8 @@ def get_qwen_text(uploaded_file, model, processor):
             generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
         )
         return output_text
-    
+
+@st.cache_data
 # Streamlit app title
 st.title("OCR Image Text Extraction")