Update app.py

app.py

@@ -118,16 +118,12 @@ with ui.navset_card_tab(id="tab"):
             ax.set_ylabel("Loss rate")
             return fig
 
-        @render.image
+        @render.plot()
         def plot_context_size_scaling():
             df = pd.read_csv("14m.csv")
             fig = plot_loss_rates(df, "14M")
             if fig:
-                import tempfile
-
-                fd, path = tempfile.mkstemp(suffix=".svg")
-                fig.savefig(path)
-                return {"src": str(path), "width": "600px", "format": "svg"}
+                return fig
 
     with ui.nav_panel("Model loss analysis"):
         ui.panel_title("Neurips stuff")
@@ -178,7 +174,7 @@ with ui.navset_card_tab(id="tab"):
             ax.set_ylabel("Loss rate")
             return fig
 
-        @render.image
+        @render.plot()
         def plot_model_scaling():
             df = pd.read_csv("training_data_5.csv")
             df = df[df["epoch_interp"] > 0.035]
@@ -186,11 +182,7 @@ with ui.navset_card_tab(id="tab"):
                 df, input.param_type(), input.loss_type(), input.model_type()
             )
             if fig:
-                import tempfile
-
-                fd, path = tempfile.mkstemp(suffix=".svg")
-                fig.savefig(path)
-                return {"src": str(path), "width": "600px", "format": "svg"}
+                return fig
 
     with ui.nav_panel("Scaling Laws"):
         ui.panel_title("Params & Losses")
@@ -237,15 +229,11 @@ with ui.navset_card_tab(id="tab"):
             ax.set_ylabel("Loss")
             return fig
 
-        @render.image
+        @render.plot()
         def plot_big_boy_model():
             df = pd.read_csv("training_data_5.csv")
             fig = plot_loss_rates_model_scale(
                 df, input.loss_type_scale(), input.model_type_scale()
             )
             if fig:
-                import tempfile
-
-                fd, path = tempfile.mkstemp(suffix=".svg")
-                fig.savefig(path)
-                return {"src": str(path), "width": "600px", "format": "svg"}
+                return fig