From 17a5eb0cbed4d6583341fdcffdaa4df9324c871d Mon Sep 17 00:00:00 2001
From: Shrimad Mishra <42738198+shrimad-mishra@users.noreply.github.com>
Date: Wed, 7 Feb 2024 21:39:37 +0530
Subject: [PATCH] added a return statement to fix predict functionality

---
 deployement/working/IndicLID.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deployement/working/IndicLID.py b/deployement/working/IndicLID.py
index 1157335..a7a3d89 100644
--- a/deployement/working/IndicLID.py
+++ b/deployement/working/IndicLID.py
@@ -297,7 +297,7 @@ def get_dataloaders(self, indices, input_texts, batch_size):
         
     def predict(self, input):
         input_list = [input,]
-        self.batch_predict(input_list, 1)
+        return self.batch_predict(input_list, 1)
 
     def batch_predict(self, input_list, batch_size):