Upload app.py

app.py

@@ -10,6 +10,12 @@ import pandas as pd
 from huggingface_hub import CommitScheduler
 from pathlib import Path
 
+import subprocess
+
+# Run the training script
+subprocess.run(["python", "train.py"])
+print('done!')
+
 # Configure the logging functionality
 log_file = Path("logs/") / f"data_{uuid.uuid4()}.json"
 log_folder = log_file.parent
@@ -63,7 +69,7 @@ def predict_insurance_charges(age, bmi, children, sex, smoker, region):
 
     data_point = pd.DataFrame([sample])
     
-    #prediction = insurance_charge_predictor.predict(data_point).tolist()
+    prediction = insurance_charge_predictor.predict(data_point).tolist()
 
     with scheduler.lock:
         with log_file.open("a") as f:
@@ -101,8 +107,4 @@ gr_interface.queue()
 gr_interface.launch(share=False)
 
 print('*** Running train.py ***')
-import subprocess
 
-# Run the training script
-subprocess.run(["python", "train.py"])
-print('*** done! ***')