Update helpers.py

helpers.py

@@ -46,7 +46,10 @@ def recognition(imgs, ids, names, source_faces, d, source_imgs):
               axes[i, j].set_axis_off()
               img_count+=1
             else:
-              axes[i, j].set_title(f"Roll No.:{source_imgs[ids[img_count]].split('/')[-1].split('.')[0]}")
-              axes[i, j].imshow(source_faces[ids[img_count]])
+              if names[img_count] == "Unknown":
+                axes[i, j].set_title(f"Unknown")
+              else:
+                axes[i, j].set_title(f"Roll No.:{source_imgs[ids[img_count]].split('/')[-1].split('.')[0]}")
+                axes[i, j].imshow(source_faces[ids[img_count]])
               axes[i, j].set_axis_off()
   plt.savefig("Recognition.jpg")