Fixing issue

app.py

@@ -57,16 +57,17 @@ st.caption("For more detail: [Github link](https://github.com/SmithaUpadhyaya/fa
 
 #Select few sample images for the catagory of cloths
 st.caption("Select image:")
+option = 'None'
 option = st.selectbox('From sample', ('None', 'dress', 'earrings', 'sweater', 'sunglasses', 'shoe', 'hat', 'heels', 'socks', 'tee', 'bracelet'), index = 0)
 st.text("Or")
 file_name = st.file_uploader(label = "Upload an image", accept_multiple_files = False)
 
+image = None
 btn_click = st.button('Generate')
 st.caption("Application deployed on CPU basic with 16GB RAM")
 
 if btn_click:
 
-    image = None
     if file_name is not None:     
 
         image = Image.open(file_name)
@@ -78,6 +79,11 @@ if btn_click:
 
     if image is not None:
 
+        image_col, caption_text = st.columns(2)
+        image_col.header("Image")
+        caption_text.header("Generated Caption")
+        image_col.image(image.resize((252,252)), use_column_width = True)
+
         if 'init_model_required' not in st.session_state:
             with st.spinner('Initializing model...'):
 
@@ -93,11 +99,7 @@ if btn_click:
             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.resize((252,252)), use_column_width = True)
+        with st.spinner('Generating Caption...'):            
 
             #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)"
@@ -110,10 +112,9 @@ if btn_click:
 
             #Predict the caption for the imahe
             generated_ids = model.generate(pixel_values = pixel_values, max_length = 25)
-            generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]  
+            generated_caption = processor.batch_decode(generated_ids, skip_special_tokens = True)[0]  
 
-            #Output the predict text
-            caption_text.header("Generated Caption")
+            #Output the predict text            
             caption_text.text(generated_caption)