Spaces:
Running
Running
Update posts_categorization.py
Browse files- posts_categorization.py +1 -2
posts_categorization.py
CHANGED
@@ -91,8 +91,7 @@ def classify_post(summary, labels):
|
|
91 |
{"role": "system", "content": "You are a strict classifier. Return only one matching category name under the field 'category'."},
|
92 |
{"role": "user", "content": prompt}
|
93 |
],
|
94 |
-
temperature=0
|
95 |
-
max_tokens=60
|
96 |
)
|
97 |
return result.category
|
98 |
except Exception as e:
|
|
|
91 |
{"role": "system", "content": "You are a strict classifier. Return only one matching category name under the field 'category'."},
|
92 |
{"role": "user", "content": prompt}
|
93 |
],
|
94 |
+
temperature=0
|
|
|
95 |
)
|
96 |
return result.category
|
97 |
except Exception as e:
|