deve

app.py

@@ -248,7 +248,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
 def predict(text):
     config = AutoConfig.from_pretrained("shorecode/t5-efficient-tiny-nh8-summarizer")
     model = AutoModelForSeq2SeqLM.from_config(config)
-    initialize_weights(model)
+    #initialize_weights(model)
     tokenizer = AutoTokenizer.from_pretrained('shorecode/t5-efficient-tiny-nh8-summarizer')
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
     outputs = model(inputs)