Spaces:

VietAI
/

ViNewsSum

Runtime error

razent commited on May 28, 2022

Commit

58d943c

1 Parent(s): f769ccc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-large-vietnews-summarizat
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large-vietnews-summarization")
 def preprocess(inp):
-    text = "summarize: " + inp + " </s>"
     features = tokenizer(text, return_tensors="pt")
     return features['input_ids'], features['attention_mask']
 def predict(input_ids, attention_mask):

 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large-vietnews-summarization")
 def preprocess(inp):
+    text = "vietnews: " + inp + " </s>"
     features = tokenizer(text, return_tensors="pt")
     return features['input_ids'], features['attention_mask']
 def predict(input_ids, attention_mask):