Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr 
 import matplotlib.pyplot as plt
 import numpy as np
+from matplotlib.ticker import MultipleLocator
 
 INTRO = """# Harm's law 
 
@@ -48,14 +49,16 @@ for kn in kns:
     overheads.append(compute_overhead(kn, kd)*100)
 
 def plot_curve(kn, kd):
-    fig = plt.figure()
+    fig, ax = plt.subplots()
     plt.plot(kns, overheads, color="black")
     plt.scatter([kn], [compute_overhead(kn, kd)*100], marker="x", c="red", label="You are here!")
-    plt.scatter([1.0], [100.0], marker="x", c="blue", label="Chinchilla optimal")
-    plt.grid(True)
+    plt.scatter([1.0], [0.0], marker="x", c="blue", label="Chinchilla optimal")
     plt.xlabel("Fraction of compute optimal model size")
     plt.ylabel("Compute overhead (%)")
     plt.legend(loc="best")
+    plt.grid(True, which="both")
+    ax.yaxis.set_minor_locator(MultipleLocator(10))
+
     return fig