Moustapha91 commited on
Commit
5a5fa15
·
verified ·
1 Parent(s): 0d7113a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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.",