Launched app

README.md

@@ -1,12 +1,10 @@
 ---
 title: Real_estate_ie
-emoji: 🐨
+emoji: 🏡
 colorFrom: indigo
 colorTo: blue
 sdk: gradio
 sdk_version: 3.0.11
 app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+pinned: true
+---

app.py

@@ -0,0 +1,86 @@
+# all the imports
+from transformers import pipeline
+import gradio as gr
+import torch
+import re
+
+#---------------------------------------------------------------------------------
+# function modified for the demo
+def get_native_feature_demo(language, input_description, target_feature, question, return_numbers = True):
+    task = 'question-answering'
+    french_model = 'etalab-ia/camembert-base-squadFR-fquad-piaf'
+    english_model = 'deepset/roberta-base-squad2'
+    if language == "French":
+        model = pipeline(task, model=french_model, tokenizer=french_model)
+    else:
+        model = pipeline(task, model=english_model, tokenizer=english_model)
+    answer = model({
+        'question': question,
+        'context': input_description
+    })
+    if target_feature == "price":
+        # price median
+        reference_target_value = 1650
+    else:
+        # living space median
+        reference_target_value = 85
+    try:
+        if return_numbers:
+            nums = re.findall(r"\d*\.\d+|\d+\,\d+|\d{1,3}'[\d{3}']+\d{1,3}\.?\d*|(?<!m)\d+",answer['answer'])
+            nums = [re.sub("\,|\'", "", num) for num in nums]
+            # turn every number into float
+            if len(nums) > 1:
+                nums = [float(x) for x in nums]
+                answer_processed = min(nums, key=lambda x:abs(x-reference_target_value))
+            elif len(nums) == 1:
+                answer_processed = float(''.join(nums))
+        else:
+            answer = answer['answer']
+        answer_full = answer['answer']
+        score = answer['score']
+        return round(answer_processed), score, answer_full
+    except:
+        return "No answer!", None, None
+
+#---------------------------------------------------------------------------------
+# define the parameters
+title = "Extract relevant features from Real Estate Descriptions!"
+
+description = """
+<center>
+<img src="https://i0.wp.com/lightersideofrealestate.com/wp-content/uploads/2014/11/2-koalafications.jpg" width=300px>
+</center>
+<div style="text-align: justify">
+Features with missing instances can negatively impact the performance of machine learning models. Information Extraction (IE) can improve the availability of tabular data by identifying relevant information from unstructured textual descriptions. This project demonstrated the application of IE on descriptions of online real estate listings, whereby the required missing values are retrieved from the text. Inspired by question-answering tasks, the aim was to recover these values by asking a set of questions. We tested two ways to achieve this goal. The first one focuses on a model specific to the language of the description (French) to perform IE, while the second translates the descriptions into English before IE. The project compared the performance of both approaches while delivering insights on how the formulation of the questions can impact the effectiveness of Q&A models.
+</div>
+"""
+article = """
+<center>
+
+Created by [Ilia Azizi](https://iliaazizi.netlify.app/) 2022
+
+</center>
+"""
+
+interesting_case = """ Venez découvrir votre futur appartement lors des portes ouvertes Quai Ouest à Renens le samedi 6 mars de 9h00 à 12h00. Afin de respecter les mesures de sécurité, merci d'inscrire une personne par tranche horaire. Merci de laisser vos coordonnées (nom, email et téléphone) afin que l'on puisse vous joindre pour confirmer le rendez-vous et l'adresse. Lorsque vous entrer votre nom pour vous inscrire, merci d'indiquer également l'appartement que vous souhaitez visiter (n° ou nombre de pièces). En cas d'intérêt, merci de venir avec un dossier complet : - 3 dernières fiches de salaire - Extrait de l'office des poursuites - Copie carte d'identité ou permis de séjour - Copie de votre police d'assurance RC (responsabilité civile) et ménage Un formulaire de location sera à disposition sur place. Pour en savoir plus: https://quai-ouest.ch/location/logements\nINSCRIPTIONS : https://doodle.com/poll/5cvrqyt2ueaah3su?utm_source=poll&utm_medium=link\nNouvelle promotion de 89 logements en location idéalement située à la Gare de Renens, Quai Ouest ouvrira ses bras aux futurs locataires avec plusieurs options de logements, commerces et bureaux.\nCe magnifique appartement de 4.5 pièces se situe au 3° étage et a été construit avec de très beaux matériaux et se compose comme suit:\nSpacieux séjour de 40m2\nSuite parentale de 23.9m2 avec salle de douche italienne attenante\nDeux chambres à coucher d'environ 13m2 chacune\nRaccordement pour colonne de lavage\nUne salle de bains/WC\nCuisine ouverte avec îlot central et entièrement agencée avec lave-vaisselle\nStores électriques\nEntrée avec armoires murales\nLoyer net : CHF 2'600 .-\nCharges: CHF 320.-\nTotal: CHF 2'920.-\nAdresse: Place de la Gare à 1020 Renens\nPlus d'informations sur www.quai-ouest.ch/location/logements ainsi que chez Omnia Immobilier au 021 943 54 54 ou riviera@omnia.ch"""
+
+examples = [["French", interesting_case, 'price', "Combien payez-vous de loyer?"],
+["French", interesting_case, 'living_space', "Quelle est la surface en m2?"]]
+
+#---------------------------------------------------------------------------------
+# run the demo
+demo = gr.Interface(
+    fn=get_native_feature_demo,
+    inputs=[
+            gr.inputs.Radio(["French", "English"], label="Language", type = "value"),
+            gr.inputs.Textbox(lines=10, label="Real Estate Description", placeholder="Type a real estate description here."),
+            gr.inputs.Dropdown(["price", "living_space"], label="Target Feature", type = "value"),
+            gr.inputs.Textbox(lines=2, label="Question", placeholder="Ask to retrieve a feature.")],
+    outputs = [gr.outputs.Textbox(label = "Extracted Feature"), gr.outputs.Textbox(label = "Probability of the prediction"), gr.outputs.Textbox(label = "Full Q&A answer")],
+    title = title,
+    description = description,
+    article = article,
+    examples=examples
+)
+
+demo.launch(inline=False)

requirements.txt

@@ -0,0 +1,5 @@
+transformers
+torch
+regex
+gradio
+sentencepiece