Prettify OOS prediction

app.py

@@ -34,8 +34,9 @@ def predict(model_type, input):
     
 def oos_predict(input):
     inputs = tokenizer(input, return_tensors="pt")
-    outputs = model(**inputs)
-    return str(outputs)
+    outputs = model(**inputs).logits
+    outputs = torch.softmax(torch.tensor(outputs), dim=-1)[0]
+    return str({"In scope": outputs[1], "Out of scope": outputs[0]})
     
 
 def rasa_predict(input):