Update app.py

app.py

@@ -60,7 +60,7 @@ Asegúrese de que la traducción sea precisa y conserve el significado original
     
     formatted_prompt = template.replace("{TEXT}", text).replace("{LANGUAGE}", target_language)
     response = llm_engine_hf.invoke(formatted_prompt)
-    translated_text = response['generated_text']
+    translated_text = response.content
     
     return translated_text
 
@@ -74,7 +74,7 @@ Asegúrese de que el resumen sea conciso y conserve el significado original del
     '''
     
     response = llm_engine_hf.invoke(template)
-    summarized_text = response['generated_text']
+    summarized_text = response.content
     
     return summarized_text
 
@@ -105,13 +105,15 @@ def handle_uploaded_file(uploaded_file):
         return str(e)
 
 def main():
-    st.title("LexAIcon")
+    st.set_page_config(page_title="LexAIcon", page_icon="utils/icon.jpg", layout="centered", initial_sidebar_state="auto", menu_items=None)
+
+    st.image("./icon.jpg", width=100)
+
+    st.title("LexAIcon: Traduce, resume y explica textos legales")
     st.write("Puedes conversar con este chatbot basado en Mistral7B-Instruct y subir archivos para que el chatbot los procese.")
 
     if "generated" not in st.session_state:
         st.session_state["generated"] = []
-    if "past" not in st.session_state:
-        st.session_state["past"] = []
 
     # Entrada del usuario
     user_input = st.text_input("Tú: ", "")