Update app.py
Browse files
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 |
)
|