update x y axis names

app.py

@@ -64,19 +64,22 @@ def get_plot(batch_size=1024):
         results.append([s[0].shape for s in sampled])
         break
     
-    y = th.tensor(results)
+    t = th.tensor(results)
 
-    d = {"x": [], "y": []}
+    x = "sampler"
+    y = "# vertices"
+
+    d = {x: [], y: []}
 
     for i, name in enumerate(names):
-        yy = y[:, i]
-        d["y"] += yy.tolist()
-        d["x"] += [name] * yy.shape[0]
+        yy = t[:, i]
+        d[y] += yy.tolist()
+        d[x] += [name] * yy.shape[0]
     
     update = gr.BarPlot.update(
         value=pd.DataFrame(d),
-        x="x",
-        y="y",
+        x=x,
+        y=y,
         title="Number of sampled vertices",
         width=600,
         height=350