Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2
3# Load the model
4model = AutoModelForCausalLM.from_pretrained("Alihallaba/smol3-simple-helper")
5tokenizer = AutoTokenizer.from_pretrained("Alihallaba/smol3-simple-helper")
6
7# Create a pipeline for text generation
8pipe = pipeline(
9 "text-generation",
10 model=model,
11 tokenizer=tokenizer,
12 max_new_tokens=80
13)
14
15# Example usage
16prompt = "Simplify this: Explain black holes"
17output = pipe(prompt)[0]["generated_text"]
18print(output)