Update app.py

app.py

@@ -146,25 +146,25 @@ with col2:
                 st.text_area("Transcription :", transcription['text'])
                 st.session_state['transcription'] = transcription['text']
 
-
-if st.session_state['transcription'] is not None:
-    inputs = tokenizer(transcription['text'].lower(), return_tensors="pt").input_ids
-    outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
-    pred = tokenizer.decode(outputs[0], skip_special_tokens=True)
+with st.spinner("Affichage des pictogrammes..."):
+    if st.session_state['transcription'] is not None:
+        inputs = tokenizer(transcription['text'].lower(), return_tensors="pt").input_ids
+        outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
+        pred = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    sentence_to_map = process_output_trad(pred)
-    pictogram_ids = [get_id_picto_from_predicted_lemma(lexicon, lemma) for lemma in sentence_to_map]
-    st.session_state['pictogram_ids'] = [get_id_picto_from_predicted_lemma(lexicon, lemma) for lemma in sentence_to_map]
+        sentence_to_map = process_output_trad(pred)
+        pictogram_ids = [get_id_picto_from_predicted_lemma(lexicon, lemma) for lemma in sentence_to_map]
+        st.session_state['pictogram_ids'] = [get_id_picto_from_predicted_lemma(lexicon, lemma) for lemma in sentence_to_map]
 
 
-if st.session_state['pictogram_ids'] is not None:
-    html = generate_html(st.session_state['pictogram_ids'])
-    st.components.v1.html(html, height=500, scrolling=True)
+    if st.session_state['pictogram_ids'] is not None:
+        html = generate_html(st.session_state['pictogram_ids'])
+        st.components.v1.html(html, height=500, scrolling=True)
 
-    # Container to hold the download button
-    pdf_path = generate_pdf(st.session_state['pictogram_ids'])
-    with open(pdf_path, "rb") as pdf_file:
-        st.download_button(label="Télécharger la traduction en PDF", data=pdf_file, file_name="pictograms.pdf", mime="application/pdf")
+        # Container to hold the download button
+        pdf_path = generate_pdf(st.session_state['pictogram_ids'])
+        with open(pdf_path, "rb") as pdf_file:
+            st.download_button(label="Télécharger la traduction en PDF", data=pdf_file, file_name="pictograms.pdf", mime="application/pdf")
 
 # record_audio = st_audiorec()
 # if record_audio: