Spaces:
Sleeping
Sleeping
First true version
Browse files
app.py
CHANGED
@@ -2,6 +2,10 @@ import os
|
|
2 |
import re
|
3 |
import logging
|
4 |
import nltk
|
|
|
|
|
|
|
|
|
5 |
import torch
|
6 |
import gradio as gr
|
7 |
from transformers import pipeline, AutoConfig
|
@@ -20,10 +24,6 @@ model_name = "AgentPublic/camembert-base-toxic-fr-user-prompts"
|
|
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 |
-
nltk.data.find('corpora/wordnet')
|
26 |
-
nltk.download('wordnet')
|
27 |
|
28 |
lemmatizer = WordNetLemmatizer()
|
29 |
insult_words = [
|
|
|
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 |
config = AutoConfig.from_pretrained(model_name)
|
25 |
classifier = pipeline('text-classification', model=model_name, device=device)
|
26 |
|
|
|
|
|
|
|
|
|
27 |
|
28 |
lemmatizer = WordNetLemmatizer()
|
29 |
insult_words = [
|