Update app.py

app.py

@@ -47,7 +47,7 @@ torch.set_grad_enabled(False)
 
 
 def past_kv_to_device(past_kv, device):
-    return [(k.to(device).detach(), v.to(device).detach()) for k, v in past_kv.items()]
+    return tuple((k.to(device).detach(), v.to(device).detach()) for k, v in past_kv)
 
 
 @spaces.GPU