Merge pull request #153 from PromtEngineer/issue-152-Bug-fix-in-show_…

…sources-flag Issue-152 bug fix with --show_sources
PromtEngineer · Jun 15, 2023 · 28095bd · 28095bd
2 parents 8906385 + 511142d
commit 28095bd
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/run_localGPT.py b/run_localGPT.py
@@ -142,18 +142,13 @@ def load_model(device_type, model_id, model_basename=None):
 )
 @click.option(
  "--show_sources",
- default=False,
- type=click.Choice(
- [
- False,
- True,
- ]
- ),
+ "-s",
+ is_default=True,
  help="Show sources along with answers (Default is False)",
 )
 def main(device_type, show_sources):
  """
- This function implements the information retreival task.
+ This function implements the information retrieval task.
 
 
  1. Loads an embedding model, can be HuggingFaceInstructEmbeddings or HuggingFaceEmbeddings