Spaces:
Sleeping
Sleeping
Fix incorrect comment
Browse files- 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
|
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
|