Update app.py

app.py

@@ -71,9 +71,10 @@ def find_sentences(query):
                 continue
             else:
                 doc = corpus[idx]
+                return doc
 
                 break
-    return doc
+    return corpus[corpus_id]