Update app.py

app.py

@@ -20,7 +20,7 @@ def plot_forecast(num_param, batch_size, precision, seq_len):
     y3 = batch_size * seq_len * K * precision / (1024**3)
 
     # Gradients: N×precision
-    y4 = num_param * precision / (1024**3)
+    y4 = num_param * 4 / (1024**3)
 
     fig = plt.figure(figsize=(4, 4))
     ax = fig.add_subplot(111)