Spaces:

vives
/

bert_sentence_sim

Runtime error

vives commited on May 13, 2022

Commit

1c9c58f

1 Parent(s): 3bba4cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
 model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
-model_base = AutoModelForMaskedLM.from_pretrained("distilbert-base-uncased")
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 text1 = st.text_area("Enter first sentence")
 text2 = st.text_area("Enter second sentence")

 model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
 model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
+model_base = AutoModelForMaskedLM.from_pretrained("distilbert-base-uncased", output_hidden_states=True)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 text1 = st.text_area("Enter first sentence")
 text2 = st.text_area("Enter second sentence")