neek05/NLP-AMLO

1from transformers import BloomForCausalLM, AutoTokenizer2import gradio as gr3 4# Loading and setting up GPT2 Open AI Transformer Model5model_path = './model_path_BLOOM_vf'6model = BloomForCausalLM.from_pretrained(model_path)7tokenizer = AutoTokenizer.from_pretrained(model_path)8 9def generate_text(inp):10    # Encoding the starting point of the sentence we want to predict11    input_data = tokenizer.encode(inp, return_tensors='pt')12     13    # Generating Output String14    output = model.generate(15        input_data, 16        max_length= 100,  17        do_sample=True,  18        top_k=0,19        top_p=0.92,20    )21    return tokenizer.decode(output[0], skip_special_tokens=True)22 23textbox = gr.Textbox(label="Introduce una o más palabras para generar el texto:", placeholder="Por ejemplo: Los jóvenes", lines=1)    24 25gr.Interface(fn=generate_text,26             inputs= textbox, 27             outputs="text",28             examples=[29        ["Los jóvenes"],30        ["La economía ha"],31        ["Los conservadores"],32        ["Hemos trabajado para"],33        ["El crimen organizado"],34      ],35    ).launch()