add code for space

README.md

@@ -1,8 +1,8 @@
 ---
 title: Supervised Contrastive Learning
-emoji: 🏢
+emoji: 🐨
 colorFrom: yellow
-colorTo: gray
+colorTo: orange
 sdk: gradio
 app_file: app.py
 pinned: false
@@ -12,7 +12,7 @@ license: apache-2.0
 # Configuration
 
 `title`: _string_
-Display title for the Space
+Supervised Contrastive Learning
 
 `emoji`: _string_
 Space emoji (emoji-only character allowed)

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import tensorflow as tf
+from huggingface_hub import from_pretrained_keras
+
+model = from_pretrained_keras("keras-io/supervised-contrastive-learning-cifar10")
+
+labels = ["airplane", "automobile", "bird", "cat", "deer", "dog", "frog", "horse", "ship", "truck"]
+
+
+def infer(test_image):
+    image = tf.constant(test_image)
+    image = tf.reshape(image, [-1, 32, 32, 3])
+    pred = model.predict(image)
+    pred_list = pred[0, :]
+    return {labels[i]: float(pred_list[i]) for i in range(10)}
+
+
+image = gr.inputs.Image(shape=(32, 32))
+label = gr.outputs.Label(num_top_classes=3)
+
+
+article = """<center>
+Authors: <a href='https://twitter.com/johko990' target='_blank'>Johannes Kolbe</a> after an example by Khalid Salama at 
+<a href='https://keras.io/examples/vision/supervised-contrastive-learning/' target='_blank'>keras.io</a> <br>
+<a href='https://arxiv.org/abs/2004.11362' target='_blank'>Original paper</a> by Prannay Khosla et al."""
+
+
+description = """Classification with a model trained via Supervised Contrastive Learning """
+
+
+Iface = gr.Interface(
+    fn=infer,
+    inputs=image,
+    outputs=label,
+    examples=[["examples/cat.jpg"], ["examples/ship.jpeg"]],
+    title="Supervised Contrastive Learning Classification",
+    article=article,
+    description=description,
+).launch()

requirements.txt

@@ -0,0 +1,3 @@
+tensorflow >=2.6.0
+gradio
+huggingface_hub