Spaces:
Runtime error
Runtime error
Commit
·
a762485
1
Parent(s):
2f444b0
change default to camembert
Browse files
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("
|
174 |
|
175 |
|
176 |
###
|
@@ -209,12 +209,12 @@ print(extract_top_3(test_article))
|
|
209 |
|
210 |
some_examples = [
|
211 |
|
212 |
-
["
|
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 |
-
["
|
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.
|