ewan-rvl commited on
Commit
8860eb6
·
1 Parent(s): fa97403

First true version

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