Akhil Koduri commited on
Commit
d2abc66
·
verified ·
1 Parent(s): 44f8d36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -6
app.py CHANGED
@@ -29,12 +29,6 @@ def classify_text(text):
29
 
30
  label = labels[predicted_class]
31
 
32
- # Adjust classification based on score
33
- if confidence_score < 0.75:
34
- label = "Negative"
35
-
36
- return label, confidence_score
37
-
38
  # Streamlit app
39
  st.title("BERT Text Classification")
40
  st.write("This app classifies text using a pre-trained BERT model.")
 
29
 
30
  label = labels[predicted_class]
31
 
 
 
 
 
 
 
32
  # Streamlit app
33
  st.title("BERT Text Classification")
34
  st.write("This app classifies text using a pre-trained BERT model.")