Spaces:
Sleeping
Sleeping
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.",
|