Update app.py

app.py

@@ -10,17 +10,14 @@ repo_id = "isboudja/Spam"
 learner = from_pretrained_fastai(repo_id)
 labels = learner.dls.vocab
 
-# Define a function to make predictions
 def predict(text):
-    prediction = learner.predict(text)
-    pred = prediction[0]
-    probs = prediction[1:]
-    return {pred: float(probs[i]) for i, pred in enumerate(labels)}
+    pred, pred_idx, probs = learner.predict(text)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
-# Create and launch the Gradio interface
+# Create and launch the Gradio interface for text input
 gr.Interface(
     fn=predict, 
     inputs=gr.inputs.Textbox(lines=2, placeholder="Enter text here..."), 
     outputs=gr.outputs.Label(),
-    examples=["Hello, how are you?", "You won a prize! Click here to claim."]
+    examples=["WON a guaranteed £1000 cash"]
 ).launch(share=False)