Downloading models

app.py

@@ -4,21 +4,13 @@ from transformers import pipeline
 import gc
 
 # Download models
-u = pipeline('fill-mask', model='Daniel-Saeedi/Sent-Debias-bert-gender-debiased')
-del u
-gc.collect()
+bert_debiased = pipeline('fill-mask', model='Daniel-Saeedi/Sent-Debias-bert-gender-debiased')
+bert_original = pipeline('fill-mask', model='Daniel-Saeedi/Sent-Debias-bert-gender-debiased')
+
 
 def fill_mask(stmt,model):
     if model == 'bert':
-        debiased = 'Daniel-Saeedi/debiased-bert-base-uncased'
-        original = 'bert-base-uncased'
-
-    
-    unmasker_1 = pipeline('fill-mask', model=debiased)
-    
-    unmasker_2 = pipeline('fill-mask', model=original)
-    
-    return unmasker_1(stmt), unmasker_2(stmt)
+        return bert_debiased(stmt), bert_original(stmt)
 
 
 demo = gr.Interface(