Update app.py

app.py

@@ -3,9 +3,18 @@ from transformers import pipeline
 
 pipeline = pipeline(task="text-classification", model="databoyface/distilbert-base-cased-ome-v4.2")
 
-def predict(input):
+def predict(input: str):
     predictions = pipeline(input)
-    return {p["label"]: p["score"] for p in predictions}
+    result = []
+    for p in predictions:
+        try:
+            response = OMEResponse(p["label"])
+        except ValueError:
+            response = OMEResponse.INDETERMINATE
+
+        result.append(f"So, {response.reaction} from a result: {response.display_name} with score: {p['score']:.3f} .")
+
+    return "\n".join(result)
     
 gradio_app = gr.Interface(
     fn=predict,