Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ model_name = "Moustapha91/bart_large_poetique-v02"
|
|
7 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
|
|
|
10 |
# Fonction de génération ajustée
|
11 |
def generate_summary(input_text):
|
12 |
inputs = tokenizer(
|
@@ -33,6 +34,7 @@ def generate_summary(input_text):
|
|
33 |
output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
34 |
return output_str
|
35 |
|
|
|
36 |
# Liste d'exemples enrichis
|
37 |
examples = [
|
38 |
"L'amour, cet élan mystérieux qui unit les âmes, est une lumière douce dans les ténèbres du quotidien.",
|
|
|
7 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
|
10 |
+
# Fonction de génération ajustée
|
11 |
# Fonction de génération ajustée
|
12 |
def generate_summary(input_text):
|
13 |
inputs = tokenizer(
|
|
|
34 |
output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
35 |
return output_str
|
36 |
|
37 |
+
|
38 |
# Liste d'exemples enrichis
|
39 |
examples = [
|
40 |
"L'amour, cet élan mystérieux qui unit les âmes, est une lumière douce dans les ténèbres du quotidien.",
|