frankai98 commited on
Commit
e5ef478
·
verified ·
1 Parent(s): e71a06c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,7 +80,7 @@ def load_models():
80
  model="meta-llama/Llama-3.2-1B-Instruct",
81
  device=0, # Use GPU if available
82
  torch_dtype=torch.bfloat16,) # Use FP16 for efficiency
83
- st.success("Llama 3.2 summarization model loaded successfully!")
84
 
85
  # Display success message that will disappear after 5 seconds
86
  Thread(
@@ -98,7 +98,7 @@ def load_models():
98
  score_pipe = pipeline("text-classification",
99
  model="cardiffnlp/twitter-roberta-base-sentiment-latest",
100
  device=0 if torch.cuda.is_available() else -1)
101
- st.success("Sentiment analysis model loaded successfully!")
102
 
103
  # Display success message that will disappear after 5 seconds
104
  Thread(
 
80
  model="meta-llama/Llama-3.2-1B-Instruct",
81
  device=0, # Use GPU if available
82
  torch_dtype=torch.bfloat16,) # Use FP16 for efficiency
83
+ #st.success("Llama 3.2 summarization model loaded successfully!")
84
 
85
  # Display success message that will disappear after 5 seconds
86
  Thread(
 
98
  score_pipe = pipeline("text-classification",
99
  model="cardiffnlp/twitter-roberta-base-sentiment-latest",
100
  device=0 if torch.cuda.is_available() else -1)
101
+ #st.success("Sentiment analysis model loaded successfully!")
102
 
103
  # Display success message that will disappear after 5 seconds
104
  Thread(