Update app.py

app.py

@@ -2,6 +2,19 @@ import sys
 import numpy as np
 import gradio as gr
 import requests
+from accelerate import Accelerator
+
+accelerator = Accelerator(gradient_accumulation_steps=2)
+dataloader, model, optimizer, scheduler = accelerator.prepare(dataloader, model, optimizer, scheduler)
+
+with accelerator.accumulate():
+    for input, output in dataloader:
+        outputs = model(input)
+        loss = loss_func(outputs)
+        loss.backward()
+        optimizer.step()
+        scheduler.step()
+        optimizer.zero_grad()
 
 
 def greet(payload):