Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load the model and tokenizer
4model_name = "your-username/fine_tuned_llama_3_2_assignment_grader"
5model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8# Generate predictions
9input_text = "Sample student response text here."
10inputs = tokenizer(input_text, return_tensors="pt")
11outputs = model.generate(**inputs)
12print(tokenizer.decode(outputs[0]))