File size: 1,068 Bytes
817f66a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
---
{}
---
### 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))