Downloading models

app.py

@@ -17,7 +17,7 @@ def make_slider(unmask):
     html = '<div>'
 
     for word in unmask:
-        html += '<h3>{}</h3><input type="range" disabled min="0" max="100" value="{}" style="width: 100%;">'.format(word['token_str'],factor*word['score'])
+        html += '<h5>{}</h5><input type="range" disabled min="0" max="100" value="{}%" style="width: 100%;">'.format(word['token_str'],factor*word['score'])
     
     html += '</div>'
     return html
@@ -26,7 +26,7 @@ def make_slider(unmask):
 
 def fill_mask(stmt,model):
     if model == 'bert':
-        return make_slider(bert_debiased(stmt))
+        return "<h2>Debiased:</h2>" + make_slider(bert_debiased(stmt)) + "<h2>Original:</h2>" + make_slider(bert_original(stmt))
 
 
 demo = gr.Interface(