nijatzeynalov commited on
Commit
817f66a
1 Parent(s): 87f318d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +24 -0
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))