ewan-rvl commited on
Commit
fa97403
·
1 Parent(s): 845f9a1

First true version

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = [