chibuzordev commited on
Commit
d9e0463
Β·
verified Β·
1 Parent(s): 62d6099

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -29
app.py CHANGED
@@ -1,29 +1,29 @@
1
- # app.py
2
-
3
- import streamlit as st
4
- from your_pipeline_module import AdversarialAttackPipeline, RAGPipeline
5
-
6
- # Instantiate RAG and pipeline
7
- rag = RAGPipeline()
8
- pipeline = AdversarialAttackPipeline(answer_generator=rag.generate_answer)
9
-
10
- # Streamlit UI
11
- st.set_page_config(page_title="βš”οΈ Adversarial Attack Viewer", layout="wide")
12
- st.title("βš”οΈ Adversarial Attack Viewer")
13
- st.markdown("Compare original and perturbed inputs/outputs without metrics. Focus on behavior change.")
14
-
15
- # Inputs
16
- query = st.text_input("Enter Original Query")
17
- perturb_method = st.selectbox("Perturbation Method", ["synonym", "delete", "contextual"])
18
- k = st.slider("Top-K Retrieved Chunks", min_value=1, max_value=5, value=3)
19
-
20
- # Run attack
21
- if st.button("Run Attack"):
22
- with st.spinner("Running adversarial pipeline..."):
23
- result = pipeline.run(query=query, top_k=k, perturb_method=perturb_method)
24
-
25
- st.text_area("πŸ”΅ Original Query", result['normal_query'], height=100)
26
- st.text_area("🟠 Perturbed Query", result['perturbed_query'], height=100)
27
- st.text_area("🟒 Normal Response", result['normal_response'], height=150)
28
- st.text_area("🟑 Perturbed Response", result['perturbed_response'], height=150)
29
- st.text_area("πŸ”΄ Adversarial Response (Perturbed Normal Response)", result['adversarial_response'], height=150)
 
1
+ # app.py
2
+
3
+ import streamlit as st
4
+ from rag_pipeline3 import AdversarialAttackPipeline, RAGPipeline
5
+
6
+ # Instantiate RAG and pipeline
7
+ rag = RAGPipeline()
8
+ pipeline = AdversarialAttackPipeline(answer_generator=rag.generate_answer)
9
+
10
+ # Streamlit UI
11
+ st.set_page_config(page_title="βš”οΈ Adversarial Attack Viewer", layout="wide")
12
+ st.title("βš”οΈ Adversarial Attack Viewer")
13
+ st.markdown("Compare original and perturbed inputs/outputs without metrics. Focus on behavior change.")
14
+
15
+ # Inputs
16
+ query = st.text_input("Enter Original Query")
17
+ perturb_method = st.selectbox("Perturbation Method", ["synonym", "delete", "contextual"])
18
+ k = st.slider("Top-K Retrieved Chunks", min_value=1, max_value=5, value=3)
19
+
20
+ # Run attack
21
+ if st.button("Run Attack"):
22
+ with st.spinner("Running adversarial pipeline..."):
23
+ result = pipeline.run(query=query, top_k=k, perturb_method=perturb_method)
24
+
25
+ st.text_area("πŸ”΅ Original Query", result['normal_query'], height=100)
26
+ st.text_area("🟠 Perturbed Query", result['perturbed_query'], height=100)
27
+ st.text_area("🟒 Normal Response", result['normal_response'], height=150)
28
+ st.text_area("🟑 Perturbed Response", result['perturbed_response'], height=150)
29
+ st.text_area("πŸ”΄ Adversarial Response (Perturbed Normal Response)", result['adversarial_response'], height=150)