Update app.py

app.py

@@ -59,11 +59,11 @@ def get_rhymes(inp, level):
     return filtered_rhymes
 
 def get_inputs_length(input):
-  gpt2_tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
-  input_ids = gpt2_tokenizer(input)['input_ids']
-  return len(input_ids)
-  
- tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
+    gpt2_tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
+    input_ids = gpt2_tokenizer(input)['input_ids']
+    return len(input_ids)
+
+tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
 model = RobertaForMaskedLM.from_pretrained('roberta-base')
 text_generation = pipeline("text-generation")
 set_seed(0)