Change return dict value to float

app.py

@@ -2,17 +2,19 @@ import gradio as gr
 import os
 from functions import *
 
+title = "DeepLense CvT: Substructure Prediction"
 examples_dir = 'examples'
-title = "CvT: Substructure Prediction"
 
+# loading the example files
 examples = []
 for root, dirs, files in os.walk(examples_dir):
     for file in files:
         if file.endswith(".jpg"):
             examples.append([os.path.join(root, file), os.path.join(root, file)])
 
+# initialising the interface
 interface = gr.Interface(fn=predict, inputs=gr.Image(type= 'pil', shape=(256, 256), image_mode= 'L'),
-            outputs= gr.Label(num_top_classes=2), cache_examples= False,
+            outputs= gr.Label(), cache_examples= False,
             examples= examples, title= title, css= '.gr-box {background-color: rgb(230 230 230);}')
 
 interface.launch()

functions.py

@@ -14,5 +14,5 @@ def predict(img):
     pred = session.run([output_name], {input_name: img})[0]
     pred = np.exp(pred) / np.sum(np.exp(pred), axis=1, keepdims=True)
 
-    class_probs = {'No Substructure': str(pred[0][0]), 'Substructure': str(pred[0][1])}
+    class_probs = {'No Substructure': float(pred[0][0]), 'Substructure': float(pred[0][1])}
     return class_probs