Update app.py

app.py

@@ -9,7 +9,7 @@
 # - Custom Sam2Config inherits from PretrainedConfig to solve subscriptable errors.
 # - UI polling is backward-compatible with older Gradio versions.
 # -------------------------------
-
+import time
 import math
 import json
 import requests
@@ -381,6 +381,7 @@ def run_tuning_task():
         print("Starting model tuning on CPU...")
         peft_model.train()
         for epoch in range(NUM_EPOCHS):
+            time.wait(0.01)
             for i, batch in enumerate(train_dataloader):
                 input_ids = batch['input_ids'].to(device)
                 attention_mask = batch['attention_mask'].to(device)