Update app.py

app.py

@@ -6,20 +6,20 @@ import json
 import gradio as gr
 import pandas as pd
 
-# from huggingface_hub import CommitScheduler
+from huggingface_hub import CommitScheduler
 from pathlib import Path
 
 
-# log_file = Path("logs/") / f"data_{uuid.uuid4()}.json"
-# log_folder = log_file.parent
+log_file = Path("logs/") / f"data_{uuid.uuid4()}.json"
+log_folder = log_file.parent
 
-# scheduler = CommitScheduler(
-#     repo_id="machine-failure-logs",
-#     repo_type="dataset",
-#     folder_path=log_folder,
-#     path_in_repo="data",
-#     every=2
-# )
+scheduler = CommitScheduler(
+    repo_id="machine-failure-logs",
+    repo_type="dataset",
+    folder_path=log_folder,
+    path_in_repo="data",
+    every=2
+)
 
 machine_failure_predictor = joblib.load('model.joblib')
 
@@ -47,20 +47,20 @@ def predict_machine_failure(air_temperature, process_temperature, rotational_spe
     data_point = pd.DataFrame([sample])
     prediction = machine_failure_predictor.predict(data_point).tolist()
 
-    # with scheduler.lock:
-    #     with log_file.open("a") as f:
-    #         f.write(json.dumps(
-    #             {
-    #                 'Air temperature [K]': air_temperature,
-    #                 'Process temperature [K]': process_temperature,
-    #                 'Rotational speed [rpm]': rotational_speed,
-    #                 'Torque [Nm]': torque,
-    #                 'Tool wear [min]': tool_wear,
-    #                 'Type': type,
-    #                 'prediction': prediction[0]
-    #             }
-    #         ))
-    #         f.write("\n")
+    with scheduler.lock:
+        with log_file.open("a") as f:
+            f.write(json.dumps(
+                {
+                    'Air temperature [K]': air_temperature,
+                    'Process temperature [K]': process_temperature,
+                    'Rotational speed [rpm]': rotational_speed,
+                    'Torque [Nm]': torque,
+                    'Tool wear [min]': tool_wear,
+                    'Type': type,
+                    'prediction': prediction[0]
+                }
+            ))
+            f.write("\n")
             
     return prediction[0]
 
@@ -72,7 +72,7 @@ demo = gr.Interface(
     title="Machine Failure Predictor",
     description="This API allows you to predict the machine failure status of an equipment",
     allow_flagging="auto",
-    concurrency_limit=8
+    concurrency_limit=16
 )
 
 demo.queue()