Update app.py
Browse files
app.py
CHANGED
@@ -114,7 +114,7 @@ def rag_process(query, k=2):
|
|
114 |
|
115 |
return response, retrieved_faqs, metrics
|
116 |
|
117 |
-
# Plot RAG pipeline
|
118 |
def plot_metrics(metrics):
|
119 |
data = pd.DataFrame({
|
120 |
'Stage': ['Embedding', 'Retrieval', 'Generation'],
|
@@ -127,13 +127,13 @@ def plot_metrics(metrics):
|
|
127 |
sns.set_palette("muted")
|
128 |
|
129 |
ax1 = sns.barplot(x='Stage', y='Latency (ms)', data=data, color='skyblue')
|
130 |
-
ax1.set_ylabel('Latency (ms)', color='
|
131 |
-
ax1.tick_params(axis='y', labelcolor='
|
132 |
|
133 |
ax2 = ax1.twinx()
|
134 |
-
sns.lineplot(x='Stage', y='Accuracy (%)', data=data, marker='o', color='
|
135 |
-
ax2.set_ylabel('Accuracy (%)', color='
|
136 |
-
ax2.tick_params(axis='y', labelcolor='
|
137 |
|
138 |
plt.title('RAG Pipeline: Latency and Accuracy')
|
139 |
plt.tight_layout()
|
@@ -219,13 +219,13 @@ body {
|
|
219 |
background: #303030;
|
220 |
border-radius: 8px;
|
221 |
align-items: center;
|
222 |
-
width:
|
223 |
}
|
224 |
#output-container {
|
225 |
background: #303030;
|
226 |
padding: 15px;
|
227 |
border-radius: 8px;
|
228 |
-
width:
|
229 |
}
|
230 |
.text-center {
|
231 |
text-align: center;
|
|
|
114 |
|
115 |
return response, retrieved_faqs, metrics
|
116 |
|
117 |
+
# Plot RAG pipeline (remove red color)
|
118 |
def plot_metrics(metrics):
|
119 |
data = pd.DataFrame({
|
120 |
'Stage': ['Embedding', 'Retrieval', 'Generation'],
|
|
|
127 |
sns.set_palette("muted")
|
128 |
|
129 |
ax1 = sns.barplot(x='Stage', y='Latency (ms)', data=data, color='skyblue')
|
130 |
+
ax1.set_ylabel('Latency (ms)', color='skyblue')
|
131 |
+
ax1.tick_params(axis='y', labelcolor='skyblue')
|
132 |
|
133 |
ax2 = ax1.twinx()
|
134 |
+
sns.lineplot(x='Stage', y='Accuracy (%)', data=data, marker='o', color='lightblue', linewidth=2)
|
135 |
+
ax2.set_ylabel('Accuracy (%)', color='lightblue')
|
136 |
+
ax2.tick_params(axis='y', labelcolor='lightblue')
|
137 |
|
138 |
plt.title('RAG Pipeline: Latency and Accuracy')
|
139 |
plt.tight_layout()
|
|
|
219 |
background: #303030;
|
220 |
border-radius: 8px;
|
221 |
align-items: center;
|
222 |
+
width: 33%;
|
223 |
}
|
224 |
#output-container {
|
225 |
background: #303030;
|
226 |
padding: 15px;
|
227 |
border-radius: 8px;
|
228 |
+
width: 66%;
|
229 |
}
|
230 |
.text-center {
|
231 |
text-align: center;
|