Hugging Face's logo

Spaces:
vives
/
bert_sentence_sim
Runtime error

App Files Community
vives commited on
Commit
4bd311a
•
1 Parent(s): 821e07e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from sklearn.metrics.pairwise import cosine_similarity
4
  import streamlit as st
5
  import torch
6
 
7
- model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
8
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
9
  model_base = AutoModelForMaskedLM.from_pretrained("distilbert-base-uncased", output_hidden_states=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
4
  import streamlit as st
5
  import torch
6
 
7
+ model_checkpoint = "vives/distilbert-base-uncased-finetuned-cvent-2022"
8
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
9
  model_base = AutoModelForMaskedLM.from_pretrained("distilbert-base-uncased", output_hidden_states=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)