Create example

README.md

@@ -1,12 +1,14 @@
 ---
-title: Hartmann6D
-emoji: 💻
-colorFrom: purple
-colorTo: indigo
+title: Hartmann-6
+emoji: 📈
+colorFrom: yellow
+colorTo: purple
 sdk: gradio
-sdk_version: 4.23.0
+sdk_version: 4.21.0
 app_file: app.py
 pinned: false
+license: mit
+short_description: Optimize a six-dimensional transfer learning example
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+from botorch.test_functions.synthetic import Hartmann
+from torch import Tensor
+
+
+def hartmann(x1, x2, x3, x4, x5, x6):
+    x_tensor = Tensor((x1, x2, x3, x4, x5, x6))
+    hartmann_function = Hartmann(dim=6)
+    y = hartmann_function(x_tensor)
+
+    return float(y)
+
+
+iface = gr.Interface(
+    fn=hartmann,
+    inputs=[
+        gr.Number(0.33, label="x1", minimum=0.0, maximum=1.0),
+        gr.Number(0.33, label="x2", minimum=0.0, maximum=1.0),
+        gr.Number(0.33, label="x3", minimum=0.0, maximum=1.0),
+        gr.Number(0.33, label="x4", minimum=0.0, maximum=1.0),
+        gr.Number(0.33, label="x5", minimum=0.0, maximum=1.0),
+        gr.Number(0.33, label="x6", minimum=0.0, maximum=1.0),
+    ],
+    outputs=gr.Number(
+        hartmann(0.33, 0.33, 0.33, 0.33, 0.33, 0.33), label="Hartmann function value"
+    ),
+)
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+numpy
+gradio
+botorch
+torch