Spaces:
Sleeping
Sleeping
First true version
Browse files
app.py
CHANGED
@@ -2,10 +2,6 @@ import os
|
|
2 |
import re
|
3 |
import logging
|
4 |
import nltk
|
5 |
-
# Chargement des ressources NLTK
|
6 |
-
nltk.download('punkt')
|
7 |
-
nltk.data.find('corpora/wordnet')
|
8 |
-
nltk.download('wordnet')
|
9 |
import torch
|
10 |
import gradio as gr
|
11 |
from transformers import pipeline, AutoConfig
|
@@ -24,6 +20,12 @@ model_name = "AgentPublic/camembert-base-toxic-fr-user-prompts"
|
|
24 |
config = AutoConfig.from_pretrained(model_name)
|
25 |
classifier = pipeline('text-classification', model=model_name, device=device)
|
26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
27 |
|
28 |
lemmatizer = WordNetLemmatizer()
|
29 |
insult_words = [
|
|
|
2 |
import re
|
3 |
import logging
|
4 |
import nltk
|
|
|
|
|
|
|
|
|
5 |
import torch
|
6 |
import gradio as gr
|
7 |
from transformers import pipeline, AutoConfig
|
|
|
20 |
config = AutoConfig.from_pretrained(model_name)
|
21 |
classifier = pipeline('text-classification', model=model_name, device=device)
|
22 |
|
23 |
+
# Chargement des ressources NLTK
|
24 |
+
nltk.download('punkt')
|
25 |
+
try:
|
26 |
+
nltk.data.find('corpora/wordnet')
|
27 |
+
except LookupError:
|
28 |
+
nltk.download('wordnet')
|
29 |
|
30 |
lemmatizer = WordNetLemmatizer()
|
31 |
insult_words = [
|