Save init model in session state

Save the init model in the session state to prevent model init again when try with another images

app.py

@@ -23,7 +23,6 @@ map_sampleid_name = {
                     'tee': 'e2d8637a-5478-429d-a2a8-3d5859dbc64d.jpeg',
                     'bracelet': 'e78518ac-0f54-4483-a233-fad6511f0b86.jpeg'
                     }
-init_model_required = True
 
 def init_model(init_model_required):
 
@@ -79,14 +78,26 @@ if btn_click:
 
     if image is not None:
 
-        with st.spinner('Initializing model...'):
-            processor, model, init_model_required = init_model(init_model_required)
+        if 'init_model_required' not in st.session_state:
+            with st.spinner('Initializing model...'):
+
+                init_model_required = True
+                processor, model, init_model_required = init_model(init_model_required)
+
+                #Save session init model in session state
+                if 'init_model_required' not in st.session_state:
+                    st.session_state.init_model_required = init_model_required
+                    st.session_state.processor = processor
+                    st.session_state.model = model
+        else:
+            processor = st.session_state.processor
+            model = st.session_state.model
 
         with st.spinner('Generating Caption...'):
 
             image_col, caption_text = st.columns(2)
             image_col.header("Image")
-            image_col.image(image, use_column_width = True)
+            image_col.image(image.resize((252,252)), use_column_width = True)
 
             #Preprocess the image
             #Inferance on GPU. When used this on GPU will get errors like: "slow_conv2d_cpu" not implemented for 'Half'" , " Input type (float) and bias type (struct c10::Half)"