AdithyaSK commited on
Commit
b3edf84
·
1 Parent(s): bb60941

Update README.md and app.py: change SDK version and modify heatmap figure creation

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 👁️
4
  colorFrom: yellow
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 4.44.0
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
4
  colorFrom: yellow
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 5.9.1
8
  app_file: app.py
9
  pinned: false
10
  license: mit
app.py CHANGED
@@ -259,7 +259,7 @@ def generate_colgemma_heatmap(
259
  )
260
 
261
  # Create matplotlib figure
262
- fig, ax = plt.subplots(figsize=(10, 10))
263
  ax.imshow(img_array)
264
  ax.imshow(
265
  similarity_map_image,
 
259
  )
260
 
261
  # Create matplotlib figure
262
+ _, ax = plt.subplots(figsize=(10, 10))
263
  ax.imshow(img_array)
264
  ax.imshow(
265
  similarity_map_image,