Update app.py

app.py

@@ -17,14 +17,20 @@ def get_model_output(input_text):
     }
     #output = query({"inputs": "The answer to the universe is <mask>.", })
     output = query(payload)
-    return output
+    results = []
+    for item in output:
+        sequence = item.get('sequence', '')
+        score = item.get('score', 0)
+        results.append(f"{sequence} (Score: {score:.4f})")
+    return "\n".join(results)
+    
 
 # Define Gradio interface
 demo = gr.Interface(
     fn=get_model_output,
-    inputs=gr.Textbox(lines=2, placeholder="Enter a sentence with <mask> to fill..."),
+    inputs=gr.Textbox(lines=2, label="Input with mask token", placeholder="Enter a sentence with <mask> to fill..."),
     outputs="json",  # Display the output as JSON to inspect the response structure
-    title="nasa-smd-ibm-v0.1 Model Output",
+    title="nasa-smd-ibm-v0.1 Model Output (Fill Mask)",
     description="Input a sentence with <mask> to see the model's predictions."
 )