Update app.py

app.py

@@ -51,24 +51,28 @@ model, tokenizer = load_model()
 # -------------------------------
 # Interfaz de usuario
 # -------------------------------
-st.title("✈️ Predicción de Rating para Reseñas de Aerolíneas")
 
-airlines = ["Iberia", "Vueling", "Ryanair", "Air Europa", "EasyJet", "Eurowings","Grupo AirFrance-KLM","Grupo IAG","Iberia Express",
-           "Jet2.com","Lufthansa","Norwegian","Pegasus Airlines","SAS","Turkish Airlines","Wizz Air"]
-selected_airline = st.selectbox("Selecciona una aerolínea", airlines)
+st.title("✈️ Airline Review Rating Prediction")
 
-review = st.text_area("Escribe tu reseña de vuelo aquí:")
+airlines = ["Iberia", "Vueling", "Ryanair", "Air Europa", "EasyJet", "Eurowings",
+            "Grupo AirFrance-KLM", "Grupo IAG", "Iberia Express", "Jet2.com",
+            "Lufthansa", "Norwegian", "Pegasus Airlines", "SAS", "Turkish Airlines", "Wizz Air"]
 
-if st.button("Predecir Rating"):
+selected_airline = st.selectbox("Select an airline", airlines)
+
+review = st.text_area("Write your flight review here:")
+
+if st.button("Predict Rating"):
     inputs = tokenizer(review, return_tensors="pt", truncation=True, padding=True, max_length=256)
     with torch.no_grad():
         output = model(**inputs).squeeze()
-        rating = output.item() * 1.65  # Ajuste de escala
+        rating = output.item() * 1.65  # Scale adjustment
         rating = min(rating, 10)
         rating = round(rating, 2)
 
-    with st.expander(f"Resultados para {selected_airline}"):
-        st.success(f"🌟 Rating predicho: **{rating}/10**")
+    with st.expander(f"Results for {selected_airline}"):
+        st.success(f"🌟 Predicted Rating: **{rating}/10**")
         mostrar_analisis_aerolinea(df_reviews, selected_airline)
 
 
+