ak3ra commited on
Commit
70691a9
1 Parent(s): 9a9bac9

structuring

Browse files
Files changed (1) hide show
  1. app.py +0 -8
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",