fix: JSON o/p instead of label

app.py

@@ -13,18 +13,18 @@ def predict(text):
   outputs = model(**inputs)
   probs = outputs.logits.softmax(dim=-1)
   pred_index = probs.argmax(dim=-1).item()
-  confidence_score = probs[0, pred_index]
+  confidence_score = probs[0, pred_index].item()
   id2label = model.config.id2label
   pred_label = id2label[pred_index]
 
-  return {'topic': pred_label, 'confidence': confidence_score}
+  return {'topic': pred_label, 'confidence': round(confidence_score, 4)}
   
 title = "URL content Topic Categorizer"
 
 topic = gr.Interface(
   fn=predict, 
   inputs='text',
-  outputs='label',
+  outputs= gr.JSON(),
   title=title,
 )