stupid reverse reverse stuff for sanity checking

app.py

@@ -14,6 +14,8 @@ model_name.choices = ["English to Hebrew", "Hebrew to English"]
 # An output text box displaying the translated text and reverse translated text
 translation = gr.outputs.Textbox(label="Translation")
 reverse_translation = gr.outputs.Textbox(label="Reverse Translation")
+reverse_reverse_translation = gr.outputs.Textbox(label="Reverse Reverse Translation")
+reverse_reverse_reverse_translation = gr.outputs.Textbox(label="Reverse Reverse Reverse Translation")
 
 # A function for translating text
 def translate(model_name, text):
@@ -57,9 +59,20 @@ def translate(model_name, text):
     reverse_outputs = reverse_model.generate(reverse_input_ids)
     reverse_translation = reverse_tokenizer.decode(reverse_outputs[0], skip_special_tokens=True)
     
-    return forward_translation, reverse_translation
+    # Reverse Reverse translation
+    reverse_reverse_input_ids = forward_tokenizer.encode(reverse_translation, return_tensors="pt")
+    reverse_reverse_outputs = forward_model.generate(reverse_reverse_input_ids)
+    reverse_reverse_translation = forward_tokenizer.decode(reverse_reverse_outputs[0], skip_special_tokens=True)
 
-iface = gr.Interface(fn=translate, inputs=[model_name, "text"], outputs=[translation, reverse_translation])
+    # Reverse Reverse Reverse translation
+    reverse_reverse_reverse_input_ids = reverse_tokenizer.encode(reverse_reverse_translation, return_tensors="pt")
+    reverse_reverse_reverse_outputs = reverse_model.generate(reverse_reverse_reverse_input_ids)
+    reverse_reverse_reverse_translation = reverse_tokenizer.decode(reverse_reverse_reverse_outputs[0], skip_special_tokens=True)
+    
+    # Return the translation
+    return forward_translation, reverse_translation, reverse_reverse_translation, reverse_reverse_reverse_translation
+    
+iface = gr.Interface(fn=translate, inputs=[model_name, "text"], outputs=[translation, reverse_translation, reverse_reverse_translation, reverse_reverse_reverse_translation])
 
 # Launch the interface
 iface.launch(share=False)