Update back.py

back.py

@@ -2,7 +2,7 @@ import streamlit as st
 import numpy as np
 import torch
 
-from typing import TypeVar
+from typing import TypeVar, Tuple
 from transformers import DistilBertTokenizer
 
 
@@ -42,7 +42,7 @@ label_mapping = {
 }
 
 
-def load_setup(path_to_model: str, path_to_vocab: str) -> tuple[ModelType, TokenizerType]:
+def load_setup(path_to_model: str, path_to_vocab: str) -> Tuple[ModelType, TokenizerType]:
     loaded_model = torch.load(path_to_model, map_location=device)
     loaded_tokenizer = DistilBertTokenizer(path_to_vocab)
     return loaded_model, loaded_tokenizer