Spaces:
Runtime error
Runtime error
File size: 454 Bytes
46e7aa4 |
1 2 3 4 5 6 7 8 9 10 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
def model_loading(modelo):
return pipeline("text-generation", model=modelo, trust_remote_code=True)
def data_processing(text, pipe, RAG=False, Adv_prompts=False, max_len=100):
prompt = f"Question: {text}.\nAnswer:"
generated_text = pipe(prompt, max_len=100, num_return_sequences=1, truncation=True)
return generated_text[0]["generated_text"].split("\n")[1].strip()
|