frankjosh commited on
Commit
693093c
·
verified ·
1 Parent(s): ca5a024

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,8 +36,9 @@ if 'feedback' not in st.session_state:
36
  st.session_state.feedback = {}
37
 
38
 
 
39
  @st.cache_data
40
- def generate_embedding(model, tokenizer, text):
41
  inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
42
  with torch.no_grad():
43
  outputs = model(**inputs)
 
36
  st.session_state.feedback = {}
37
 
38
 
39
+
40
  @st.cache_data
41
+ def generate_embedding(_model, tokenizer, text):
42
  inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
43
  with torch.no_grad():
44
  outputs = model(**inputs)