A fine-tuned Mistral 7B model specialized for business case study generation.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4# Load model
5model_name = "afzalur/case-study-mistral-7b-full"
6model = AutoModelForCausalLM.from_pretrained(
7 model_name,
8 torch_dtype=torch.float16,
9 device_map="auto"
10)
11tokenizer = AutoTokenizer.from_pretrained(model_name)
12
13# Generate case study
14prompt = "Create a case study about sustainable business practices for an MBA course"
15inputs = tokenizer(prompt, return_tensors="pt")
16outputs = model.generate(
17 **inputs,
18 max_new_tokens=512,
19 temperature=0.7,
20 do_sample=True
21)
22
23result = tokenizer.decode(outputs[0], skip_special_tokens=True)
24print(result[len(prompt):]) # Only show generated content
1# For programmatic access
2from transformers import pipeline
3
4generator = pipeline(
5 "text-generation",
6 model="afzalur/case-study-mistral-7b-full",
7 torch_dtype=torch.float16,
8 device_map="auto"
9)
10
11result = generator(
12 "Create a case study about digital transformation",
13 max_new_tokens=512,
14 temperature=0.7
15)
16print(result[0]['generated_text'])