Update app.py

app.py

@@ -17,17 +17,18 @@ def process_image(i):
     return {
         "file_path": image_path,
         "labels": imagenet_hard_dataset[i]["english_label"],
+        "english_label": imagenet_hard_dataset[i]["english_label"],
     }
 
 
 def create_fiftyone_sample(sample):
-    classifications = [
-        fo.Classification(label=str(label)) for label in sample["labels"]
-    ]
-    return fo.Sample(
-        filepath=sample["file_path"],
-        labels=fo.Classifications(classifications=classifications),
-    )
+   origin_label = fo.Classification(label=str(sample["origin"]))
+   english_label = fo.Classification(label=str(sample["english_label"]))
+   return fo.Sample(
+       filepath=sample["file_path"],
+       labels=fo.Classifications(classifications=[origin_label, english_label]),
+   )
+
 
 
 if __name__ == "__main__":