Update app.py

app.py

@@ -5,7 +5,7 @@ from typing import TypedDict
 from abc import ABC, abstractmethod
 from typing import Any, Dict, Type
 from scipy.sparse._csc import csc_matrix
-from typing import Iterable, Callable, List, Dict, Optional, Type, TypeVar
+from typing import Iterable, Callable, List, Dict, Optional, Type, TypeVar, Enum 
 import pickle
 from dataclasses import dataclass
 import tqdm
@@ -18,6 +18,7 @@ import math
 from dataclasses import dataclass
 from typing import Optional
 from datasets import load_dataset
+from enum import Enum
 
 @dataclass
 class Document: