Update app.py

app.py

@@ -22,8 +22,10 @@ def chunks_of_900(text, chunk_size=900):
     chunks = []
     current_chunk = ""
     for sentence in sentences:
-        if len(current_chunk + sentence) <= chunk_size:
-            current_chunk += " " + sentence
+        if len(current_chunk)!=0:
+            current_chunk += " "+sentence
+        else:
+            current_chunk += sentence
         else:
             chunks.append(current_chunk)
             current_chunk = sentence
@@ -53,9 +55,11 @@ def findRealProb(text):
         results.append([output, len(chunk)])
     
     ans = 0
+    cnt = 0
     for prob, length in results:
+        cnt += length
         ans = ans + prob*length
-    realProb = ans/len(text)
+    realProb = ans/cnt
     return {"Real": realProb, "Fake": 1-realProb}, results
 
 demo = gr.Interface(