docker

app.py

@@ -1,4 +1,3 @@
-import os
 import random
 import gradio as gr
 from optimum_benchmark.task_utils import (
@@ -16,9 +15,6 @@ from config_store import (
     get_pytorch_config,
 )
 
-os.system("curl -fsSL https://get.docker.com -o get-docker.sh")
-os.system("sh get-docker.sh")
-
 BACKENDS = ["pytorch", "onnxruntime", "openvino", "neural-compressor", "text-generation-inference"]
 BENCHMARKS = ["inference", "training"]
 DEVICES = ["cpu", "cuda"]

packages.txt

@@ -0,0 +1 @@
+docker.io