Update app.py

app.py

@@ -44,7 +44,7 @@ def predict(query, device="cpu"):
         probs = logits.softmax(dim=-1)
 
     fake, real = probs.detach().cpu().flatten().numpy().tolist()
-    return {"Real": real, "Fake": fake}
+    return real
 
 def findRealProb(text):
     chunksOfText = (chunks_of_600(text))
@@ -57,7 +57,7 @@ def findRealProb(text):
     for prob, length in results:
         ans = ans + prob*length
     realProb = ans/len(text)
-    return realProb
+    return {"Real": real, "Fake": 1-real}
 
 demo = gr.Interface(
         fn=findRealProb,