fixed error

app.py

@@ -4,7 +4,7 @@ import pickle
 import pandas as pd
 import gradio as gr
 
-def main(address :str, floor_area_sqft :int, remaining_lease_years :int, flat_type, storey_range, year_sold=2020 :int, update_model=False):
+def main(address :str, floor_area_sqft :int, remaining_lease_years :int, flat_type, storey_range):
     
     model_file = 'model.sav'
 
@@ -41,9 +41,6 @@ def main(address :str, floor_area_sqft :int, remaining_lease_years :int, flat_ty
     elif storey_range == 'HIGH FLOOR':
         storey_high = 1,
     
-    # loading model
-    # if update_model:
-    #    utils.get_update(year_sold)
     if os.path.exists("./model.sav"):
         model = pickle.load(open(model_file, 'rb'))
         input_dict = pd.DataFrame({
@@ -63,7 +60,7 @@ def main(address :str, floor_area_sqft :int, remaining_lease_years :int, flat_ty
             'storey_range_Mid Floor': storey_mid}, 
             index = [0])
             
-        return ('Predicted PSF:', round(model.predict(input_dict).item(),2), '\nPredicted Property Value:', round(model.predict(input_dict).item(),2) * floor_area_sqft)
+        return ({'Predicted PSF': round(model.predict(input_dict).item(),2), 'Predicted Price': round(model.predict(input_dict).item() * floor_area_sqft, 2)})
 
     else:
         return ('ERROR: No saved model')
@@ -76,8 +73,6 @@ iface = gr.Interface(
         gr.inputs.Number(default=None, label='Remaining Lease (years)', optional=False),
         gr.inputs.Dropdown(choices=['1 ROOM', '2 ROOM', '3 ROOM', '4 ROOM', '5 ROOM', 'EXECUTIVE', 'MULTI-GENERATION'], type="value", default=None, label="Flat Type"),
         gr.inputs.Dropdown(choices=['LOW FLOOR', 'MID FLOOR', 'HIGH FLOOR'], type="value", default=None, label="Storey Range"),
-        # gr.inputs.Number(default=2020, label='Data from:', optional=False),
-        # gr.inputs.Checkbox(default=False, label="Update Model?", optional=False)
     ],
     outputs = [gr.outputs.Textbox(type="auto", label='Predicted Price per SQFT')])
 iface.launch()