Update app.py

app.py

@@ -24,6 +24,10 @@ block_size = 1024
 
 model = None
 tokenizer = None
+pres_dict = {}
+
+with open('pres_dict.pkl', 'rb') as file:
+    pres_dict = pickle.load(file)
 
 def initialize():
     global model, tokenizer
@@ -77,5 +81,5 @@ def predict(request: TextInput):
         
     _, predicted = torch.max(output.data, 1)
     
-    return {"predicted": predicted.tolist()}
+    return {"predicted": pres_dict[predicted.tolist()[0]]}