dev

app.py

@@ -245,7 +245,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
 
 # Define Gradio interface
 @spaces.GPU
-def predict(text):    
+def predict(text):
+    print(text)
     config = AutoConfig.from_pretrained("shorecode/t5-efficient-tiny-nh8-summarizer")
     model = AutoModelForSeq2SeqLM.from_config(config)
     #initialize_weights(model)