Spaces:
Running
on
Zero
Running
on
Zero
Update README.md and app.py: change SDK version and modify heatmap figure creation
Browse files
README.md
CHANGED
|
@@ -4,7 +4,7 @@ emoji: 👁️
|
|
| 4 |
colorFrom: yellow
|
| 5 |
colorTo: purple
|
| 6 |
sdk: gradio
|
| 7 |
-
sdk_version:
|
| 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 |
-
|
| 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,
|