Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,21 +2,23 @@ import os
 from transformers import pipeline
 import streamlit as st
 
-port = int(os.environ.get("PORT", 7860))
+'''port = int(os.environ.get("PORT", 7860))
 st.set_page_config(page_title="Text Classifier")
 st.write(f"Running on port {port} (for Hugging Face)")
+'''
 
 # Use local cache directory to avoid permission issues
-os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
+##os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
 
 # Create zero-shot classification pipeline
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+##classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
 # Streamlit UI
+
 st.title("Zero-Shot Text Classifier")
 text = st.text_area("Enter text to classify")
 labels = st.text_input("Enter candidate labels (comma-separated)", "finance, education, health")
-
+'''
 if st.button("Classify"):
     if text and labels:
         label_list = [l.strip() for l in labels.split(",")]
@@ -24,3 +26,4 @@ if st.button("Classify"):
         st.write(result)
     else:
         st.warning("Please enter both text and labels.")
+'''