first update

app.py

@@ -19,7 +19,7 @@ def yolo(im, size=640):
 
     r= results.pred[0].numpy().T
     #probs = results.pred[0].numpy().T[-1]
-    res = [(results.names[x].lower(),p) for x,p in zip(r[-1].astype(int),r[-2].numpy())]
+    res = [(results.names[x].lower(),p) for x,p in zip(r[-1].astype(int),r[-2])]
     f_res  = story_model(res)
     #results.render()  # updates results.imgs with boxes and labels
     #return Image.fromarray(results.imgs[0])

story.py

@@ -31,6 +31,6 @@ def story_model(preds,res_len=5):
     for word,prob in preds:
         dists += np.array([prob * (word in sent) for sent in All_sentenses])
 
-    f_res = list(zip(dists,All_sentenses)).sort(key=lambda x:x[1])[-res_len:]
+    f_res = list(zip(dists,All_sentenses)).sort(key=lambda x:x[0])[-res_len:]
 
-    return f_res # list of sentenses
+    return [f for _,f in f_res] # list of sentenses