dev

app.py

@@ -249,7 +249,7 @@ def predict(text):
     config = AutoConfig.from_pretrained("shorecode/t5-efficient-tiny-nh8-summarizer")
     model = AutoModelForSeq2SeqLM.from_config(config)
     #initialize_weights(model)
-    tokenizer = AutoTokenizer.from_pretrained('shorecode/t5-efficient-tiny-nh8-summarizer')
+    tokenizer = AutoTokenizer.from_pretrained('shorecode/t5-efficient-tiny-nh8')
     inputs = tokenizer(text, return_tensors="pt", padding='max_length', max_length=512, truncation=True)
 
     # Move model and inputs to GPU if available