Update app.py

app.py

@@ -9,10 +9,10 @@ torch.set_num_interop_threads(1)
 tokenizer = T5Tokenizer.from_pretrained("cssupport/t5-small-awesome-text-to-sql")
 model = T5ForConditionalGeneration.from_pretrained("cssupport/t5-small-awesome-text-to-sql")
 # Esquema de base de datos
-SCHEMA = """
+SCHEMA_EN = """
 Database schema:
-Table bodegas(Id, Nombre, Encargado, Telefono, Email, Direccion, Horario, Regional, Latitud, Longitud)
-Table maestra(CodigoSap, Descripcion, Grupo, Agrupador, Marca, Parte, Operacion, Componente)
+Table bodegas(Id, Name, Manager, Phone, Email, Address, Schedule, Region, Latitude, Longitude)
+Table master(CodigoSap, Description, Group, Aggregator, Brand, Part, Operation, Component)
 """
 
 # Función principal
@@ -21,7 +21,7 @@ def generar_sql(pregunta_espanol):
        # Traducir pregunta a inglés
        pregunta_ingles = GoogleTranslator(source="es", target="en").translate(pregunta_espanol)
        # Crear prompt
-       prompt = f"{SCHEMA}\ntranslate English to SQL: {pregunta_ingles}"
+       prompt = f"Given the following schema:\n{SCHEMA_EN}\nTranslate this question into SQL: {pregunta_espanol}"
        input_ids = tokenizer(prompt, return_tensors="pt").input_ids
        output = model.generate(input_ids, max_length=128)
        sql = tokenizer.decode(output[0], skip_special_tokens=True)