Spaces:

joaogante
/

tf_xla_generate_benchmarks

Running

joaogante HF staff commited on Jul 15, 2022

Commit

7fa6af3

•

1 Parent(s): 9a8c083

playing around with gradio

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ demo = gr.Blocks()
 with demo:
     with gr.Tabs():
         with gr.TabItem("Greedy Search"):
-            gr.Button("New Lion")
         with gr.TabItem("Sample"):
             gr.Button("New Tiger")
         with gr.TabItem("Beam Search"):
@@ -41,9 +41,9 @@ with demo:
                     ["Transformers Version", "4.22.dev0"],
                     ["TensorFlow Version", "2.9.1"],
                     ["Pytorch Version", "1.11.0"],
-                    ["OS", "22.04 LTS (3090) / Debian 10 (others)"],
-                    ["CUDA", "11.6 (3090) / 11.3 (others)"],
-                    ["Is there code?", "Yes -- https://gist.github.com/gante/f0017e3f13ac11b0c02e4e4db351f52f"],
                 ],
             )

 with demo:
     with gr.Tabs():
         with gr.TabItem("Greedy Search"):
+            gr.Dropdown(["DistilGPT2", "GPT2", "OPT 1.3B", "GPTJ-6B", "T5 small", "T5 base", "T5 large", "T5 3B"])
         with gr.TabItem("Sample"):
             gr.Button("New Tiger")
         with gr.TabItem("Beam Search"):
                     ["Transformers Version", "4.22.dev0"],
                     ["TensorFlow Version", "2.9.1"],
                     ["Pytorch Version", "1.11.0"],
+                    ["OS", "22.04 LTS (3090) / Debian 10 (other GPUs)"],
+                    ["CUDA", "11.6 (3090) / 11.3 (others GPUs)"],
+                    ["Is there code to reproduce?", "Yes -- https://gist.github.com/gante/f0017e3f13ac11b0c02e4e4db351f52f"],
                 ],
             )