refactor installs

app.py

@@ -1,4 +1,21 @@
-import torch
+import os
+
+nvidia_available = os.system("nvidia-smi") == 0
+if nvidia_available:
+    os.system(
+        "pip install optimum-benchmark[onnxruntime-gpu,openvino,neural-compressor,diffusers,peft]@git+https://github.com/huggingface/optimum-benchmark.git"
+    )
+    DEVICES = ["cpu", "cuda"]
+else:
+    os.system(
+        "pip install optimum-benchmark[onnxruntime,openvino,neural-compressor,diffusers,peft]@git+https://github.com/huggingface/optimum-benchmark.git"
+    )
+    DEVICES = ["cpu"]
+
+BACKENDS = ["pytorch", "onnxruntime", "openvino", "neural-compressor"]
+BENCHMARKS = ["inference", "training"]
+
+
 import random
 import gradio as gr
 from optimum_benchmark.task_utils import (
@@ -15,13 +32,6 @@ from config_store import (
     get_pytorch_config,
 )
 
-cuda_available = torch.cuda.is_available()
-
-BACKENDS = ["pytorch", "onnxruntime", "openvino", "neural-compressor"]
-DEVICES = ["cpu", "cuda"] if cuda_available else ["cpu"]
-BENCHMARKS = ["inference", "training"]
-
-
 with gr.Blocks() as demo:
     # add image
     gr.Markdown(
@@ -33,11 +43,10 @@ with gr.Blocks() as demo:
     # explanation text
     gr.HTML(
         "<h3 style='text-align: center'>"
-        "A zero code Gradio interface of <a href='https://github.com/huggingface/optimum-benchmark.git'>Optimum-Benchmark</a> :<br>"
+        "Zero code Gradio interface of <a href='https://github.com/huggingface/optimum-benchmark.git'>Optimum-Benchmark</a><br>"
         "</h3>"
         "<p style='text-align: center'>"
-        "A unified multi-backend utility for benchmarking any model on the Hub including Transformers, Diffusers, PEFT and Timm models with "
-        "Optimum's arsenal of optimizations & quantization schemes, for both inference & training, on multiple backends & hardware."
+        "Unlock "
         "</p>"
     )
 

requirements.txt

@@ -1,3 +1 @@
-docker==6.1.3
-ansi2html==1.8.0
-optimum-benchmark[onnxruntime,openvino,neural-compressor,diffusers,peft]@git+https://github.com/huggingface/optimum-benchmark.git
+ansi2html==1.8.0