edodso2 commited on
Commit
8a0b401
·
1 Parent(s): a6a1c75

Fix incorrect comment

Browse files
Files changed (1) hide show
  1. utils/classify.py +1 -1
utils/classify.py CHANGED
@@ -71,7 +71,7 @@ def get_functions_for_descriptions(descriptions):
71
  Returns:
72
  dict: Mapping of descriptions to corresponding mock data functions
73
  """
74
- # Create pipeline with Facebook's BART model for zero-shot classification
75
  pipe = pipeline(model="MoritzLaurer/DeBERTa-v3-xsmall-mnli-fever-anli-ling-binary")
76
 
77
  # Call pipeline with descriptions and available Faker function labels
 
71
  Returns:
72
  dict: Mapping of descriptions to corresponding mock data functions
73
  """
74
+ # Create pipeline with small zero-shot classification model
75
  pipe = pipeline(model="MoritzLaurer/DeBERTa-v3-xsmall-mnli-fever-anli-ling-binary")
76
 
77
  # Call pipeline with descriptions and available Faker function labels