EricYuhui commited on
Commit
36ea603
·
verified ·
1 Parent(s): 583a375

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -9,7 +9,7 @@ subprocess.run([
9
  "sentence-transformers",
10
  "gdown",
11
  "matplotlib",
12
- "tf-keras" # 添加 tf-keras 到依赖列表
13
  ], check=True)
14
 
15
  import streamlit as st
@@ -223,6 +223,12 @@ def cosine_similarity(x, y):
223
  3. Return exponentiated cosine similarity
224
  (20 pts)
225
  """
 
 
 
 
 
 
226
  cosine_sim = np.dot(x, y) / (la.norm(x) * la.norm(y))
227
  return np.exp(cosine_sim)
228
 
 
9
  "sentence-transformers",
10
  "gdown",
11
  "matplotlib",
12
+ "tf-keras" # Add tf-keras to Dependency
13
  ], check=True)
14
 
15
  import streamlit as st
 
223
  3. Return exponentiated cosine similarity
224
  (20 pts)
225
  """
226
+ # Potential Division by Zero in cosine_similarity
227
+ norm_x = la.norm(x)
228
+ norm_y = la.norm(y)
229
+ if norm_x == 0 or norm_y == 0:
230
+ return 0 # Return 0 instead of NaN
231
+
232
  cosine_sim = np.dot(x, y) / (la.norm(x) * la.norm(y))
233
  return np.exp(cosine_sim)
234