yonghan93 commited on
Commit
2c33c1f
·
verified ·
1 Parent(s): 7ca3155

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,9 +3,8 @@ from pytorch_tabular.tabular_datamodule import TabularDatamodule
3
  from streamlit_option_menu import option_menu
4
  import google.generativeai as genai
5
  import os
6
- from googletrans import Translator
7
 
8
- translator = Translator()
9
 
10
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
11
 
@@ -124,7 +123,7 @@ if selected == "Prediction Tool":
124
  result = "❌ At Risk of CVD" if proba >= threshold else " ✅ Low Risk"
125
 
126
  with st.container():
127
- st.markdown("### 🧬 Prediction Result")
128
  st.markdown(f"<div style='text-align:center; font-size:26px'>{result}</div>", unsafe_allow_html=True)
129
  # st.write(f"(Probability: {proba:.2%}, Threshold: {threshold:.2f})")
130
 
 
3
  from streamlit_option_menu import option_menu
4
  import google.generativeai as genai
5
  import os
6
+ from deep_translator import GoogleTranslator
7
 
 
8
 
9
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
10
 
 
123
  result = "❌ At Risk of CVD" if proba >= threshold else " ✅ Low Risk"
124
 
125
  with st.container():
126
+ st.markdown("### 🧬Prediction Result🧬")
127
  st.markdown(f"<div style='text-align:center; font-size:26px'>{result}</div>", unsafe_allow_html=True)
128
  # st.write(f"(Probability: {proba:.2%}, Threshold: {threshold:.2f})")
129