update dropout to 0.1 for all task heads

handler.py

@@ -20,10 +20,10 @@ class EndpointHandler:
         self.model.eval()
 
         self.task_heads = torch.nn.ModuleDict({
-            'sentiment': TaskClassificationHead(self.model.config.hidden_size, 4, 0.2),
-            'topic': TaskClassificationHead(self.model.config.hidden_size, 10, 0.2),
-            'hate_speech': TaskClassificationHead(self.model.config.hidden_size, 5, 0.2),
-            'clickbait': TaskClassificationHead(self.model.config.hidden_size, 2, 0.2)
+            'sentiment': TaskClassificationHead(self.model.config.hidden_size, 4, 0.1),
+            'topic': TaskClassificationHead(self.model.config.hidden_size, 10, 0.1),
+            'hate_speech': TaskClassificationHead(self.model.config.hidden_size, 5, 0.1),
+            'clickbait': TaskClassificationHead(self.model.config.hidden_size, 2, 0.1)
         })
         self.log_vars = torch.nn.ParameterDict({
             task: torch.nn.Parameter(torch.zeros(1)) for task in self.task_heads