Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "Maxes-drew/drew-complete-legal-ai",
5 torch_dtype=torch.float16,
6 device_map="auto"
7)
8
9tokenizer = AutoTokenizer.from_pretrained("Maxes-drew/drew-complete-legal-ai")
10
11# Legal consultation example
12prompt = "Analyze this contract for potential risks:"
13inputs = tokenizer(prompt, return_tensors="pt")
14outputs = model.generate(**inputs, max_new_tokens=500)
15response = tokenizer.decode(outputs[0], skip_special_tokens=True)