|
|
|
|
|
|
|
|
from .model_loader import load_model |
|
|
from .logging_config import logger |
|
|
|
|
|
def check_if_property_related(text): |
|
|
try: |
|
|
classifier = load_model("zero-shot-classification", "typeform/mobilebert-uncased-mnli") |
|
|
result = classifier(text[:1000], ["property-related", "non-property-related"]) |
|
|
is_related = result['labels'][0] == "property-related" |
|
|
return { |
|
|
'is_related': is_related, |
|
|
'confidence': float(result['scores'][0]) |
|
|
} |
|
|
except Exception as e: |
|
|
logger.error(f"Error checking property relation: {str(e)}") |
|
|
return { |
|
|
'is_related': False, |
|
|
'confidence': 0.0 |
|
|
} |
|
|
|
