CarlosMalaga commited on
Commit
595b5fb
1 Parent(s): f3faf65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -171,8 +171,8 @@ def load_model():
171
  )
172
 
173
  retriever_outcome = GoldenRetriever(
174
- question_encoder="/home/user/app/models/retriever/question_encoder",
175
- document_index="/home/user/app/models/retriever/document_index/outcomes"
176
 
177
  )
178
 
@@ -189,8 +189,8 @@ def load_model():
189
  )
190
 
191
  retriever_outcome_db = GoldenRetriever(
192
- question_encoder="/home/user/app/models/retriever/question_encoder",
193
- document_index="/home/user/app/models/retriever/document_index/outcomes_db"
194
 
195
  )
196
 
@@ -239,14 +239,14 @@ def run_client():
239
  # Radio button selection
240
  analysis_type = st.radio(
241
  "Choose analysis type:",
242
- options=["intervention", "outcome", "question", "db intervention", "db outcome", "db question"],
243
- index=2 # Default to 'question'
244
  )
245
 
246
  # text input
247
  text = st.text_area(
248
  "Enter Text Below:",
249
- value="How does unconditional cash transver affect to reduce poverty?",
250
  height=200,
251
  max_chars=1500,
252
  )
 
171
  )
172
 
173
  retriever_outcome = GoldenRetriever(
174
+ question_encoder="models/retriever/level-4-small-no-negative-outcomes/question_encoder",
175
+ document_index="models/retriever/level-4-small-no-negative-outcomes/document_index"
176
 
177
  )
178
 
 
189
  )
190
 
191
  retriever_outcome_db = GoldenRetriever(
192
+ question_encoder="models/retriever/level-4-small-no-negative-outcomes/question_encoder",
193
+ document_index="models/retriever/level-4-small-no-negative-outcomes/document_index_db"
194
 
195
  )
196
 
 
239
  # Radio button selection
240
  analysis_type = st.radio(
241
  "Choose analysis type:",
242
+ options=["question", "intervention", "outcome", "db intervention", "db outcome"],
243
+ index=0 # Default to 'question'
244
  )
245
 
246
  # text input
247
  text = st.text_area(
248
  "Enter Text Below:",
249
+ value="How does unconditional cash transfer affect to reduce poverty?",
250
  height=200,
251
  max_chars=1500,
252
  )