utils/interpret_lab_results.py

from transformers import pipeline

generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")

def analyze_lab_results(input_text):
    prompt = (
        "Actúa como un experto en cannabis medicinal. Dado el siguiente análisis de laboratorio "
        "de una planta de cannabis, describe los efectos que podría tener en los usuarios, posibles "
        "usos terapéuticos y advertencias si las hay:\n\n"
        f"{input_text}\n\n"
        "Análisis:"
    )
    result = generator(prompt, max_length=1000, do_sample=True)
    return result[0]['generated_text'].split("Análisis:")[-1].strip()