davanstrien HF Staff commited on
Commit
9a546d7
·
verified ·
1 Parent(s): 7906865

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def _generate_summary_gpu(card_text: str, card_type: str) -> str:
80
  prefix = "<MODEL_CARD>" if card_type == "model" else "<DATASET_CARD>"
81
 
82
  # Format input according to the chat template
83
- messages = [{"role": "user", "content": f"{prefix}{card_text}"}]
84
  inputs = tokenizer.apply_chat_template(
85
  messages, add_generation_prompt=True, return_tensors="pt"
86
  )
 
80
  prefix = "<MODEL_CARD>" if card_type == "model" else "<DATASET_CARD>"
81
 
82
  # Format input according to the chat template
83
+ messages = [{"role": "user", "content": f"{prefix}{card_text[:5000]}"}]
84
  inputs = tokenizer.apply_chat_template(
85
  messages, add_generation_prompt=True, return_tensors="pt"
86
  )