jvillar02 commited on
Commit
d04c52c
·
verified ·
1 Parent(s): c157987

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -101,9 +101,7 @@ def predict(text):
101
  return f"# {pred_label}", badge_html, class_probs
102
 
103
  # --- 5. UI LAYOUT (gr.Blocks) ---
104
- # Using Soft theme (requires newer Gradio version in requirements.txt)
105
- # If it fails, remove theme=gr.themes.Soft()
106
- with gr.Blocks(theme=gr.themes.Soft()) as demo:
107
 
108
  gr.Markdown("# 📰 NLP News Classifier")
109
  gr.Markdown("Classify news articles into World, Sports, Business, or Sci/Tech using DistilBERT + LoRA.")
 
101
  return f"# {pred_label}", badge_html, class_probs
102
 
103
  # --- 5. UI LAYOUT (gr.Blocks) ---
104
+ with gr.Blocks() as demo:
 
 
105
 
106
  gr.Markdown("# 📰 NLP News Classifier")
107
  gr.Markdown("Classify news articles into World, Sports, Business, or Sci/Tech using DistilBERT + LoRA.")