nouamanetazi HF Staff commited on
Commit
4fab954
·
1 Parent(s): d51b632
Files changed (2) hide show
  1. README.md +3 -2
  2. utils.py +2 -0
README.md CHANGED
@@ -1,12 +1,13 @@
1
  ---
2
  title: Predict Memory
3
- emoji: 🐨
4
  colorFrom: indigo
5
  colorTo: blue
6
  sdk: gradio
7
  sdk_version: 5.12.0
8
  app_file: app.py
9
- pinned: false
 
10
  ---
11
 
12
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
  title: Predict Memory
3
+ emoji: 🧮
4
  colorFrom: indigo
5
  colorTo: blue
6
  sdk: gradio
7
  sdk_version: 5.12.0
8
  app_file: app.py
9
+ pinned: true
10
+ header: mini
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
utils.py CHANGED
@@ -190,6 +190,8 @@ def plot_memory_breakdown(
190
  ax2.set_xticklabels(names, rotation=45, ha='right')
191
  ax2.set_ylabel('Memory (MiB)')
192
  ax2.set_title('Aggregate Memory Metrics', pad=20)
 
 
193
 
194
  # Adjust layout to prevent text overlap
195
  plt.tight_layout()
 
190
  ax2.set_xticklabels(names, rotation=45, ha='right')
191
  ax2.set_ylabel('Memory (MiB)')
192
  ax2.set_title('Aggregate Memory Metrics', pad=20)
193
+ max_y_value = max(values)
194
+ ax2.set_ylim(0, max(80000, max_y_value))
195
 
196
  # Adjust layout to prevent text overlap
197
  plt.tight_layout()