Update app.py

app.py

@@ -11,9 +11,8 @@ def predict(text):
   THRESHOLD = prompting.compute_tokens_prob(prompt, token_list1=["good"], token_list2= ["bad"])[0].item()
   res=prompting.compute_tokens_prob(text+prompt, token_list1=["good"], token_list2= ["bad"])
   if res[0] > THRESHOLD:
-    return {"POSITIVE":(res[0].item()-THRESHOLD)/ (1-THRESHOLD)}
-  return {"NEGATIVE":(THRESHOLD-res[0].item())/THRESHOLD}
-
+    return {"POSITIVE":(res[0].item()-THRESHOLD)/ (1-THRESHOLD)}, (res[0].item()-THRESHOLD)/ (1-THRESHOLD)
+  return {"NEGATIVE":(THRESHOLD-res[0].item())/THRESHOLD},(THRESHOLD-res[0].item())/THRESHOLD  
 
 import gradio as gr
 iface = gr.Interface(fn=predict, inputs=["text"], outputs=["label"]).launch()