VicidiLochi commited on
Commit
631dead
·
1 Parent(s): 033a601

Add anonymization app 3

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # Load a lightweight model
5
- generator = pipeline("text2text-generation", model="t5-small", tokenizer="t5-small")
6
 
7
  # Function to construct the anonymization prompt
8
  def construct_prompt(input_text):
@@ -19,7 +19,7 @@ def construct_prompt(input_text):
19
  }}
20
 
21
  ### Guidelines:
22
- 1. Only replace entity names.
23
  2. Ensure the pseudonyms are realistic and appropriate for the type of name being replaced.
24
  3. Keep all other parts of the text unchanged.
25
  4. If no names are found, return the original text with an empty dictionary.
 
2
  from transformers import pipeline
3
 
4
  # Load a lightweight model
5
+ generator = pipeline("text2text-generation", model="facebook/bart-large", tokenizer="facebook/bart-large")
6
 
7
  # Function to construct the anonymization prompt
8
  def construct_prompt(input_text):
 
19
  }}
20
 
21
  ### Guidelines:
22
+ 1. Only replace entity or personnal names.
23
  2. Ensure the pseudonyms are realistic and appropriate for the type of name being replaced.
24
  3. Keep all other parts of the text unchanged.
25
  4. If no names are found, return the original text with an empty dictionary.