reformat code

inference.py

@@ -1,12 +1,8 @@
-from transformers import T5ForConditionalGeneration, T5Tokenizer
-from transformers import AutoTokenizer, MT5Model
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from transformers import MT5ForConditionalGeneration, AutoTokenizer
+from transformers import T5Tokenizer, MT5ForConditionalGeneration
 from simpletransformers.t5 import T5Model
 import datetime
 import logging
 import os
-import sys
 
 
 class Inference:
@@ -40,7 +36,6 @@ class Inference:
         max_length=256,
         device="cpu",
     ):
-      
         model = MT5ForConditionalGeneration.from_pretrained(model_name)
         tokenizer = T5Tokenizer.from_pretrained(model_name)
 
@@ -95,13 +90,8 @@ class Inference:
 def main():
     i = Inference()
     sentence = "Aquesta és una associació sense ànim de lucre amb la missió de fomentar la presència i l'ús del català."
-    #sentence = "Softcatalà és una associació sense ànim de lucre amb la missió de fomentar la presència i l'ús del català en tots els àmbits de les noves tecnologies."
-    model =  os.getcwd()
-    options = i.get_paraphrases(
-        model,
-        sentence,
-        1.0
-    )
+    model = os.getcwd()
+    options = i.get_paraphrases(model, sentence, 1.0)
     print(f"original: {sentence}")
     for option in options:
         print(f" {option}")