Update app.py

app.py

@@ -14,13 +14,16 @@ classifier = pipeline("text-classification", model="ICILS/xlm-r-icils-ilo", devi
 # Define the prediction function
 @spaces.GPU
 def classify_text(text):
-    return classifier(text)[0]
+    result = classifier(text)[0]
+    label = result['label']
+    score = result['score']
+    return label, score
 
 # Create the Gradio interface
 demo = gr.Interface(
     fn=classify_text,
     inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
-    outputs=gr.Text(),
+    outputs=[gr.Textbox(label="Label"), gr.Number(label="Score")],
     title="XLM-R ISCO classification with ZeroGPU",
     description="Classify occupations using a pre-trained XLM-R-ISCO model on Hugging Face Spaces with ZeroGPU"
 )