Description: Topic extraction from a news article and title
Original dataset: https://huggingface.co/datasets/fancyzhx/dbpedia_14
---
Try querying this adapter for free in Lora Land at https://predibase.com/lora-land!
The adapter_category is Topic Identification and the name is News Topic Identification (dbpedia)
---
Sample input: You are given the title and the body of an article below. Please determine the type of the article.\n### Title: Great White Whale\n\n### Body: Great White Whale is the debut album by the Canadian rock band Secret and Whisper. The album was in the works for about a year and was released on February 12 2008. A music video was shot in Pittsburgh for the album's first single XOXOXO. The album reached number 17 on iTunes's top 100 albums in its first week on sale.\n\n### Article Type:
---
Sample output: 11
---
Try using this adapter yourself!
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "mistralai/Mistral-7B-v0.1"
peft_model_id = "predibase/dbpedia"
model = AutoModelForCausalLM.from_pretrained(model_id)
model.load_adapter(peft_model_id)
- Downloads last month
- 2,599
Model tree for predibase/dbpedia
Base model
mistralai/Mistral-7B-v0.1