Real Estate Listing Analysis
Description: Perform various tasks to analyze real estate listings, including categorizing them by type, determining if they are for new buildings or business centers, identifying amenities, classifying listings based on location, and analyzing pricing trends.
How to Use
Here is how to use this model to classify text into different categories:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name = "interneuronai/real_estate_listing_analysis_bart"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
def classify_text(text):
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
outputs = model(**inputs)
predictions = outputs.logits.argmax(-1)
return predictions.item()
text = "Your text here"
print("Category:", classify_text(text))
- Downloads last month
- 20