bug fixes

app.py

@@ -37,16 +37,16 @@ with gr.Blocks() as demo:
         lines=5, 
         label="Input",
         type='text',
-        # rtl=True,
-        # text_align='right',
+        rtl=True,
+        text_align='right',
     )
 
     output_txt = gr.Textbox(
         lines=5, 
         label="Output",
         type='text',
-        # rtl=True,  
-        # text_align='right',
+        rtl=True,  
+        text_align='right',
     )
 
     btn = gr.Button(value="Shakkel")

dataloader.py

@@ -24,7 +24,7 @@ class DataRetriever(Dataset):
 
     def __getitem__(self, idx):
         word_x, char_x, diac_x, diac_y = self.create_sentence(idx)
-        return self.preprocess((word_x, char_x, diac_x)), T.tensor(diac_y, dtype=T.long), T.tensor(diac_y, dtype=T.long)
+        return self.preprocess((word_x, char_x, diac_x)), T.tensor(diac_y, dtype=T.long)
 
     def create_sentence(self, idx):
         line = self.lines[idx]

model_dd.py

@@ -179,7 +179,7 @@ class DiacritizerD2(nn.Module):
         for inputs, _ in tqdm(dataloader, total=len(dataloader)):
             inputs[0] = inputs[0].to(self.device)
             inputs[1] = inputs[1].to(self.device)
-            diac, _ = self(*inputs)
+            diac = self(*inputs)
 
             output = np.argmax(T.softmax(diac.detach(), dim=-1).cpu().numpy(), axis=-1)
             #^ [b ts tw]