Add application file

app.py

@@ -47,10 +47,12 @@ class Contrastive_Loss_2(tf.keras.losses.Loss):
 model = tf.keras.models.load_model( filepath='contrastive_model.h5',  custom_objects={'Contrastive_Loss_2': Contrastive_Loss_2})
 
 with open("scene_labels.json") as labels_file:
-    labels = json.load(labels_file)
+   labels = json.load(labels_file)
 
 def classify_image(img):
-    pred,idx,probs = model.predict(img)
+    img = tf.convert_to_tensor(img, dtype=tf.float32)
+    # pred,idx,probs = model.predict(img)
+    probs = model.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 image = gr.inputs.Image(shape=(256, 256))