Update app.py

app.py

@@ -185,7 +185,9 @@ def get_all_endpoints(
         endpoints_df["createdAt"] = endpoints_df["status"].apply(lambda x: x["createdAt"])
         endpoints_df["updatedAt"] = endpoints_df["status"].apply(lambda x: x["updatedAt"])
         endpoints_df = endpoints_df[["name", "model", "provider", "compute", "status", "minReplica", "maxReplica", "createdAt", "updatedAt"]]
-        return endpoints_df
+        return gr.Dataframe.update(
+            value=endpoints_df
+        )
 
 def update_endpoint(
     hf_token_input,
@@ -229,16 +231,20 @@ with gr.Blocks() as demo2:
         show_label=False,
         type="password"
     )
-    gr.Markdown("""
-    ### All Deployed Endpoints
-    """)
-
-    endpoints_table = gr.Dataframe(
-        headers=["Endpoint Name", "Revision", "Provider", "Instance Type", "Status", "Min Replica", "Max Replica", "Created At", "Updated At"],
-        col_count=(9, "fixed"),
+    with gr.Row():
+        gr.Markdown("""
+        ### All Deployed Endpoints
+        """)
+        endpoints_table = gr.Dataframe(
+            headers=["Endpoint Name", "Revision", "Provider", "Instance Type", "Status", "Min Replica", "Max Replica", "Created At", "Updated At"],
+            col_count=(9, "fixed"),
+        )
+    endpoint_info_button = gr.Button(value="Get Info")
+    endpoint_info_button.click(
+        get_all_endpoints, 
+        inputs=hf_token_input, 
+        outputs=endpoints_table
     )
-    info_button = gr.Button(value="Get Info")
-    info_button.click(get_all_endpoints, inputs=hf_token_input, outputs=endpoints_table)
 
     gr.Markdown(
     """
@@ -361,8 +367,8 @@ with gr.Blocks() as demo2:
         )
 
         security_selector = gr.Radio(
-            choices=["Protected", "Public", "Private"],
-            value="Public",
+            choices=["Protected", "Public"],
+            value="Protected",
             interactive=True,
             show_label=False,
         )