Carlosito16 commited on
Commit
a762485
·
1 Parent(s): 2f444b0

change default to camembert

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -87,7 +87,7 @@ def camembert_generate_summary(article_text):
87
  return_tensors="pt")
88
  input_ids = inputs.input_ids.to(device)
89
  attention_mask = inputs.attention_mask.to(device)
90
- output = cmb_model.generate(input_ids, attention_mask=attention_mask)
91
  return cmb_tokenizer.decode(output[0], skip_special_tokens=True)
92
 
93
 
@@ -170,7 +170,7 @@ t5_model_name = "csebuetnlp/mT5_multilingual_XLSum"
170
  t5_tokenizer = AutoTokenizer.from_pretrained(t5_model_name)
171
  t5_model = AutoModelForSeq2SeqLM.from_pretrained(t5_model_name)
172
 
173
- print(summarizer("T5", test_article))
174
 
175
 
176
  ###
@@ -209,12 +209,12 @@ print(extract_top_3(test_article))
209
 
210
  some_examples = [
211
 
212
- ["T5", """
213
  Carrefour livraison service honteux ! Ne livre jamais dans les temps, plusieurs heures de retard et si on a le malheur de travailler et de ne pas être chez nous toute la journée on peut dire au revoir à notre commande !
214
  Service client délocalisé qui insulte et raccroche au nez quand on demande ou est notre commande !
215
  Je ne recommande absolument pas, j'ai pourtant laisser plusieurs chance mais à chaque fois extrêmement déçu...
216
  """],
217
- ["T5", """
218
  Très mauvaise expérience !!!
219
  Commande livré par terre sans rien dire alors que nous étions présents.
220
  En plein centre ville de Dijon.
 
87
  return_tensors="pt")
88
  input_ids = inputs.input_ids.to(device)
89
  attention_mask = inputs.attention_mask.to(device)
90
+ output = cmb_model.generate(input_ids, attention_mask=attention_mask, max_length = 50)
91
  return cmb_tokenizer.decode(output[0], skip_special_tokens=True)
92
 
93
 
 
170
  t5_tokenizer = AutoTokenizer.from_pretrained(t5_model_name)
171
  t5_model = AutoModelForSeq2SeqLM.from_pretrained(t5_model_name)
172
 
173
+ print(summarizer("camembert", test_article))
174
 
175
 
176
  ###
 
209
 
210
  some_examples = [
211
 
212
+ ["camembert", """
213
  Carrefour livraison service honteux ! Ne livre jamais dans les temps, plusieurs heures de retard et si on a le malheur de travailler et de ne pas être chez nous toute la journée on peut dire au revoir à notre commande !
214
  Service client délocalisé qui insulte et raccroche au nez quand on demande ou est notre commande !
215
  Je ne recommande absolument pas, j'ai pourtant laisser plusieurs chance mais à chaque fois extrêmement déçu...
216
  """],
217
+ ["camembert", """
218
  Très mauvaise expérience !!!
219
  Commande livré par terre sans rien dire alors que nous étions présents.
220
  En plein centre ville de Dijon.