CCockrum commited on
Commit
80ac87f
·
verified ·
1 Parent(s): 4520c6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ if "follow_up" not in st.session_state:
23
  # --- Set Up Model & API Functions ---
24
  model_id = "mistralai/Mistral-7B-Instruct-v0.3"
25
 
26
- # Initialize sentiment analysis pipeline
27
- sentiment_analyzer = pipeline("sentiment-analysis")
28
 
29
  def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.7):
30
  return HuggingFaceEndpoint(
 
23
  # --- Set Up Model & API Functions ---
24
  model_id = "mistralai/Mistral-7B-Instruct-v0.3"
25
 
26
+ # Explicitly define the sentiment analysis model
27
+ sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english", revision="714eb0f")
28
 
29
  def get_llm_hf_inference(model_id=model_id, max_new_tokens=128, temperature=0.7):
30
  return HuggingFaceEndpoint(