File size: 1,150 Bytes
de1732e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import requests
import time
import json
from profanityfilter import ProfanityFilter
pf = ProfanityFilter()
input_text = "like"
print(pf.is_profane(input_text))
model_name = "Dabid/abusive-tagalog-profanity-detection"
endpoint = f"https://api-inference.huggingface.co/models/Dabid/abusive-tagalog-profanity-detection"
payload = {
"inputs": input_text,
"options": {
"max_length": 50,
"temperature": 1,
},
}
headers = {"Authorization": "Bearer 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"}
response = requests.post(endpoint, json=payload, headers=headers)
if response.status_code == 200:
result = json.loads(response.text)
filtered_text = result[0]
# print(filtered_text)
# print("Filtered Text:")
p = filtered_text[0]["label"]
val = filtered_text[0]["score"]
if p == "Non-Abusive" and val > 0.75:
print("Non Profane")
else:
print("Profane")
elif "Model is currently loading" in response.text:
print("Model is still loading. Retrying in a few seconds...")
time.sleep(20)
else:
print("API call failed with status code:", response.status_code)
print(response.text)
|