Update app.py

app.py

@@ -69,7 +69,7 @@ def infer(input_dataframe):
         status = "Slight congestion on E4"
     else:
         status = "Total congestion on E4"
-    return res, status
+    return pd.Dataframe[res, status]
 
 title = "Stoclholm Highway E4 Real Time Traffic Prediction"
 description = "Stockholm E4 (59°23'44.7"" N 17°59'00.4""E) highway real time traffic prediction"
@@ -89,7 +89,7 @@ with gr.Blocks() as demo:
                        # datatype=["timestamp", "float", "float", "float", "float", "float"],
                        label="Input Data", interactive=1)
         with gr.Column():
-            outputs = gr.Dataframe(row_count = (1, "fixed"), col_count=(1, "fixed"), label="Predictions", headers=["Freeflow Level"])
+            outputs = gr.Dataframe(row_count = (1, "fixed"), col_count=(2, "fixed"), label="Predictions", headers=["Freeflow Level", "Status"])
     
     with gr.Row():
         btn_sub = gr.Button(value="Submit")