Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
|
@@ -19,7 +19,7 @@ st.markdown("*Powered by DistilBERT & GitHub Models API*")
|
|
| 19 |
st.markdown("---")
|
| 20 |
|
| 21 |
# HARDCODE YOUR GITHUB API KEY HERE
|
| 22 |
-
GITHUB_TOKEN = "
|
| 23 |
|
| 24 |
# Initialize LLM client with your key
|
| 25 |
@st.cache_resource
|
|
@@ -40,7 +40,7 @@ def load_sentiment_model():
|
|
| 40 |
"""Load the fine-tuned DistilBERT model with proper 5-class handling"""
|
| 41 |
try:
|
| 42 |
# EDIT THIS: Replace with your actual model path
|
| 43 |
-
model_name = "
|
| 44 |
|
| 45 |
# Load with explicit configuration
|
| 46 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 19 |
st.markdown("---")
|
| 20 |
|
| 21 |
# HARDCODE YOUR GITHUB API KEY HERE
|
| 22 |
+
GITHUB_TOKEN = "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" # REPLACE WITH YOUR REAL KEY
|
| 23 |
|
| 24 |
# Initialize LLM client with your key
|
| 25 |
@st.cache_resource
|
|
|
|
| 40 |
"""Load the fine-tuned DistilBERT model with proper 5-class handling"""
|
| 41 |
try:
|
| 42 |
# EDIT THIS: Replace with your actual model path
|
| 43 |
+
model_name = "shivam-1706/distilbert-amazon-sentiment" # UPDATE THIS
|
| 44 |
|
| 45 |
# Load with explicit configuration
|
| 46 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|