Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 
 
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-Model = EncoderDecoderModel.from_pretrained('Bert2BertForTextDescrambling')
+Model = EncoderDecoderModel.from_pretrained('damilojohn/Bert2BertForTextDescrambling')
 def descramble(prompt):
   input = tokenizer(prompt,return_tensors='pt')
   input_id = input.input_ids
@@ -17,7 +17,7 @@ def descramble(prompt):
   return gr.Textbox.update(value=output)
  
 def set_example(example):
-    return gr.TextArea.update(value=example[0])
+  return gr.TextArea.update(value=example[0])
   
 demo = gr.Blocks()
 with demo: