Joaquin Villar
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -100,11 +100,11 @@ def predict(text):
|
|
| 100 |
st.title("📰 NLP News Classifier")
|
| 101 |
st.markdown("""
|
| 102 |
This interface uses a **DistilBERT** model fine-tuned with **LoRA (Low-Rank Adaptation)**.
|
| 103 |
-
It classifies news
|
| 104 |
""")
|
| 105 |
|
| 106 |
# Dynamic Green Banner
|
| 107 |
-
st.success(f"✅ **
|
| 108 |
|
| 109 |
text_input = st.text_area(
|
| 110 |
"Enter a News Article or Snippet:",
|
|
|
|
| 100 |
st.title("📰 NLP News Classifier")
|
| 101 |
st.markdown("""
|
| 102 |
This interface uses a **DistilBERT** model fine-tuned with **LoRA (Low-Rank Adaptation)**.
|
| 103 |
+
It classifies news headlines into four categories: **World, Sports, Business, or Sci/Tech**.
|
| 104 |
""")
|
| 105 |
|
| 106 |
# Dynamic Green Banner
|
| 107 |
+
st.success(f"✅ **Model Performance:** Accuracy: {MODEL_METRICS['Accuracy']} | F1 Score: {MODEL_METRICS['F1_Score']}")
|
| 108 |
|
| 109 |
text_input = st.text_area(
|
| 110 |
"Enter a News Article or Snippet:",
|