Views
No views yet
pip install transformers bitsandbytes peft1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3from huggingface_hub import login
4
5#Login to Huggingface to load Mistral LLM
6login("Huggingface access token")
7
8model_id = "mistralai/Mistral-7B-Instruct-v0.1"
9peft_model_name="bpavlsh/Mistral-Fake-News-Detection"
10
11tokenizer = AutoTokenizer.from_pretrained(model_id)
12base_model = AutoModelForCausalLM.from_pretrained( model_id, load_in_4bit=True,
13 device_map="auto", torch_dtype="auto")
14model = PeftModel.from_pretrained(base_model, peft_model_name)
15
16text=""" News text for analysis, from 1Kb to 10Kb """
17
18prompt = f"""<s>[INST] <<SYS>>
19You are an expert in analyzing news for fake content, propaganda, and offensive language.
20<</SYS>>
21
22Please analyze the following text: {text} [/INST]"""
23
24inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
25output = model.generate(**inputs, max_new_tokens=1500)
26output_result=tokenizer.decode(output[0], skip_special_tokens=True)
27result=output_result.split('[/INST]')[1]
28print(f"\n{result}")