Update app.py

app.py

@@ -63,7 +63,7 @@ def stream_template(template_name, **context):
     rv.disable_buffering()
     return rv
 
-def infer(query):
+def infer_gen(query):
     user_samples = split_string(query, 80, 3)
     top_k = min(5, len(descriptions))
     results = []
@@ -90,16 +90,23 @@ def infer(query):
                   specs.append(ts)
         std.extend(specs)
         results.append(sp)
-        rankings = [{ 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' },]
+        yield results
+        #rankings = [{ 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' }, { 'word': 'string', 'freq': 'count/size' },]
         #yield render_template('index.html', results=results)#, query=query, rankings=rankings)
         #yield results
         cpt += 1
     print("ranking")
     rankings = top_five_strings(std, len(user_samples))
     print("ranked")
-    return render_template('index.html', results=results, query=query, rankings=rankings)
-
+    yield render_template('index.html', results=results)#, query=query, rankings=rankings)
 
+def infer(query):
+    gen = infer_gen
+    while True:
+        try:
+            yield str(next(gen))
+        except StopIteration:
+            return
 
 
 @app.route('/', methods=['GET', 'POST'])