Update app.py

app.py

@@ -21,7 +21,10 @@ if text:
     top_n = 5
     distances = cosine_similarity(doc_embedding, candidate_embeddings)
     keywords = [candidates[index] for index in distances.argsort()[0][-top_n:]]
-    st.json("Titres d'articles correspondant à votre recherche:")
+    json_data = {
+    "message": "Titres d'articles correspondant à votre recherche:"
+    }
+    st.json(json_data)
     conn = sqlite3.connect('SU_CSV.db')
     cursor = conn.cursor()
     
@@ -45,5 +48,6 @@ if text:
     
     conn.close()
     generator = pipeline("text-generation")
-    txt=generator(doc)
-    st.json(txt)
+    txt = generator(text)[0]["generated_text"]
+
+    st.write(txt)