Update app.py

app.py

@@ -92,6 +92,7 @@ with st.form("app", clear_on_submit = True):
             image_col.header("Image")
             caption_text.header("Generated Caption")
             image_col.image(image.resize((252,252)), use_column_width = True)
+            caption_text.text("")
 
             if 'init_model_required' not in st.session_state:
                 with st.spinner('Initializing model...'):
@@ -120,7 +121,7 @@ with st.form("app", clear_on_submit = True):
                 pixel_values = inputs.pixel_values
 
                 #Predict the caption for the imahe
-                generated_ids = model.generate(pixel_values = pixel_values, max_length = 25)
+                generated_ids = model.generate(pixel_values = pixel_values, max_length = 10)
                 generated_caption = processor.batch_decode(generated_ids, skip_special_tokens = True)[0]  
 
                 #Output the predict text