Undo caching and display image with caption

app.py

@@ -19,7 +19,6 @@ def save_file_to_disk(uplaoded_file):
         f.write(uploaded_file.getbuffer())
     return temp_file
 
-@st.cache
 def load_tokenizer_and_model():
     # load the saved model
     tokenizer = AutoTokenizer.from_pretrained("dccuchile/bert-base-spanish-wwm-cased")
@@ -36,8 +35,9 @@ if uploaded_file is not None and text_input:
     local_image_path = None
     try:
         local_image_path = save_file_to_disk(uploaded_file)
-        score = run_inference(local_image_path, text_input, model, tokenizer)[0][0].tolist()
-        st.write('%s (%.2f%%)' % score)
+        score = run_inference(local_image_path, text_input, model, tokenizer).tolist()
+        st.image(uploaded_file, caption=text_input, width=None, use_column_width=None, clamp=False, channels='RGB', output_format='auto')
+        st.write(f"## Score: {score:.2f}")
     finally:
         if local_image_path:
             os.remove(local_image_path)