Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -178,7 +178,7 @@ def generate_heatmap(scores):
|
|
178 |
plt.figure(figsize=(10, 2))
|
179 |
sns.heatmap([scores], annot=True, cmap="coolwarm", xticklabels=False, yticklabels=False, cbar=True)
|
180 |
plt.title("Sentence Importance Heatmap")
|
181 |
-
plt.savefig("
|
182 |
plt.close()
|
183 |
|
184 |
# Function to highlight sentences in the summary
|
|
|
178 |
plt.figure(figsize=(10, 2))
|
179 |
sns.heatmap([scores], annot=True, cmap="coolwarm", xticklabels=False, yticklabels=False, cbar=True)
|
180 |
plt.title("Sentence Importance Heatmap")
|
181 |
+
plt.savefig("static/heatmap.png") # Save heatmap image
|
182 |
plt.close()
|
183 |
|
184 |
# Function to highlight sentences in the summary
|