nijatzeynalov
commited on
Commit
•
817f66a
1
Parent(s):
87f318d
Upload README.md with huggingface_hub
Browse files
README.md
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
{}
|
3 |
+
---
|
4 |
+
### Real Estate Listing Analysis
|
5 |
+
|
6 |
+
**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.
|
7 |
+
|
8 |
+
## How to Use
|
9 |
+
Here is how to use this model to classify text into different categories:
|
10 |
+
|
11 |
+
from transformers import AutoModelForSequenceClassification, AutoTokenizer
|
12 |
+
|
13 |
+
model_name = "interneuronai/real_estate_listing_analysis_bart"
|
14 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
15 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
16 |
+
|
17 |
+
def classify_text(text):
|
18 |
+
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
|
19 |
+
outputs = model(**inputs)
|
20 |
+
predictions = outputs.logits.argmax(-1)
|
21 |
+
return predictions.item()
|
22 |
+
|
23 |
+
text = "Your text here"
|
24 |
+
print("Category:", classify_text(text))
|