dev

app.py

@@ -251,7 +251,7 @@ def predict(text):
     model = AutoModelForSeq2SeqLM.from_config(config)
     #initialize_weights(model)
     tokenizer = AutoTokenizer.from_pretrained('shorecode/t5-efficient-tiny-nh8-summarizer')
-    inputs = tokenizer(text, padding='max_lenght', max_length=512, truncation=True)
+    inputs = tokenizer(text, padding='max_length', max_length=512, truncation=True)
     outputs = model(inputs)
     predictions = outputs.logits.argmax(dim=-1)
     return predictions.item()