Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import streamlit as st
|
|
2 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
3 |
import pandas as pd
|
4 |
from fpdf import FPDF
|
5 |
-
import time
|
6 |
|
7 |
# Interface utilisateur
|
8 |
st.set_page_config(
|
@@ -109,13 +108,11 @@ def generate_pdf(ids):
|
|
109 |
|
110 |
st.title("Traduction d'une phrase en pictogrammes ARASAAC")
|
111 |
|
112 |
-
|
113 |
-
st.toast("Text-to-Pictograms traduit une phrase en français en pictogrammes ARASAAC. Renseignez une phrase, puis validez. Vous pouvez sauvegarder la traduction au format PDF en cliquant sur le bouton en bas de page.")
|
114 |
-
time.sleep(1.5)
|
115 |
|
116 |
sentence = st.text_input("Entrez une phrase en français:")
|
117 |
-
|
118 |
-
|
119 |
inputs = tokenizer(sentence, return_tensors="pt").input_ids
|
120 |
outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
|
121 |
pred = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
3 |
import pandas as pd
|
4 |
from fpdf import FPDF
|
|
|
5 |
|
6 |
# Interface utilisateur
|
7 |
st.set_page_config(
|
|
|
108 |
|
109 |
st.title("Traduction d'une phrase en pictogrammes ARASAAC")
|
110 |
|
111 |
+
st.info("Text-to-Pictograms traduit une phrase en français en pictogrammes ARASAAC. Renseignez une phrase, puis validez. Vous pouvez sauvegarder la traduction au format PDF en cliquant sur le bouton en bas de page.", icon='ℹ️')
|
|
|
|
|
112 |
|
113 |
sentence = st.text_input("Entrez une phrase en français:")
|
114 |
+
with st.spinner("Affichage des pictogrammes..."):
|
115 |
+
if st.button('Valider'):
|
116 |
inputs = tokenizer(sentence, return_tensors="pt").input_ids
|
117 |
outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
|
118 |
pred = tokenizer.decode(outputs[0], skip_special_tokens=True)
|