Spaces:
Running
Running
structuring
Browse files
app.py
CHANGED
@@ -3,8 +3,6 @@ import json
|
|
3 |
from rag.rag_pipeline import RAGPipeline
|
4 |
from utils.prompts import highlight_prompt, evidence_based_prompt
|
5 |
from utils.prompts import (
|
6 |
-
study_characteristics_prompt,
|
7 |
-
vaccine_coverage_prompt,
|
8 |
sample_questions,
|
9 |
)
|
10 |
from config import STUDY_FILES
|
@@ -33,10 +31,6 @@ def query_rag(study_name: str, question: str, prompt_type: str) -> str:
|
|
33 |
prompt = highlight_prompt
|
34 |
elif prompt_type == "Evidence-based":
|
35 |
prompt = evidence_based_prompt
|
36 |
-
elif prompt_type == "Study Characteristics":
|
37 |
-
prompt = study_characteristics_prompt
|
38 |
-
elif prompt_type == "Vaccine Coverage":
|
39 |
-
prompt = vaccine_coverage_prompt
|
40 |
else:
|
41 |
prompt = None
|
42 |
|
@@ -112,8 +106,6 @@ with gr.Blocks() as demo:
|
|
112 |
"Default",
|
113 |
"Highlight",
|
114 |
"Evidence-based",
|
115 |
-
"Study Characteristics",
|
116 |
-
"Vaccine Coverage",
|
117 |
],
|
118 |
label="Prompt Type",
|
119 |
value="Default",
|
|
|
3 |
from rag.rag_pipeline import RAGPipeline
|
4 |
from utils.prompts import highlight_prompt, evidence_based_prompt
|
5 |
from utils.prompts import (
|
|
|
|
|
6 |
sample_questions,
|
7 |
)
|
8 |
from config import STUDY_FILES
|
|
|
31 |
prompt = highlight_prompt
|
32 |
elif prompt_type == "Evidence-based":
|
33 |
prompt = evidence_based_prompt
|
|
|
|
|
|
|
|
|
34 |
else:
|
35 |
prompt = None
|
36 |
|
|
|
106 |
"Default",
|
107 |
"Highlight",
|
108 |
"Evidence-based",
|
|
|
|
|
109 |
],
|
110 |
label="Prompt Type",
|
111 |
value="Default",
|