NikhilPatil commited on
Commit
82a0087
·
verified ·
1 Parent(s): e0c4a32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -22,13 +22,21 @@ from sentence_transformers import SentenceTransformer, CrossEncoder
22
  from langdetect import detect, detect_langs
23
  from sklearn.ensemble import IsolationForest
24
  import joblib
 
25
 
26
- # Debug: Confirm CrossEncoder import
27
  try:
28
  from sentence_transformers import CrossEncoder
29
  st.write("sentence_transformers.CrossEncoder imported successfully")
 
 
 
 
 
 
30
  except ImportError as e:
31
  st.error(f"Failed to import CrossEncoder from sentence_transformers: {str(e)}")
 
32
 
33
  # Use tf_keras imports to match environment
34
  from tf_keras.models import load_model
@@ -288,7 +296,7 @@ def predict_disease(query: str) -> Dict[str, Any]:
288
  if best_score < threshold:
289
  return {
290
  "message": {
291
- 'mr': "हा रोग आमच्या डेटाबेसमध्ये नाही.",
292
  'hi': "यह रोग हमारे डेटाबेस में नहीं है।",
293
  'en': "This disease is not in our database."
294
  }[lang],
 
22
  from langdetect import detect, detect_langs
23
  from sklearn.ensemble import IsolationForest
24
  import joblib
25
+ import pkg_resources
26
 
27
+ # Debug: Verify imports and library versions
28
  try:
29
  from sentence_transformers import CrossEncoder
30
  st.write("sentence_transformers.CrossEncoder imported successfully")
31
+ st.write(f"sentence-transformers version: {pkg_resources.get_distribution('sentence-transformers').version}")
32
+ try:
33
+ import einops
34
+ st.write(f"einops version: {pkg_resources.get_distribution('einops').version}")
35
+ except ImportError:
36
+ st.error("einops package is missing. Please ensure it is installed.")
37
  except ImportError as e:
38
  st.error(f"Failed to import CrossEncoder from sentence_transformers: {str(e)}")
39
+ st.stop()
40
 
41
  # Use tf_keras imports to match environment
42
  from tf_keras.models import load_model
 
296
  if best_score < threshold:
297
  return {
298
  "message": {
299
+ 'mr': "हा रोग आमच्या डेटाबेसमध्ये नाही。",
300
  'hi': "यह रोग हमारे डेटाबेस में नहीं है।",
301
  'en': "This disease is not in our database."
302
  }[lang],