Lin0He
/

text-summary-gpt2-short

text-generation

text-generation-inference

Model card Files Files and versions Community

Lin0He commited on Dec 5, 2023

Commit

d987ebd

·

1 Parent(s): a8bc91b

Update handler.py

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -10,6 +10,7 @@ model.push_to_hub(repo_name="text-summary-gpt2-short", repo_id="Lin0He/text-summ
 tokenizer.push_to_hub(repo_name="text-summary-gpt2-short", repo_id="Lin0He/text-summary-gpt2-short")
 '''
 import torch
 from typing import Dict, List, Any
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer

 tokenizer.push_to_hub(repo_name="text-summary-gpt2-short", repo_id="Lin0He/text-summary-gpt2-short")
 '''
 import torch
+import numpy as np
 from typing import Dict, List, Any
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer