Update app.py

app.py

@@ -1,10 +1,8 @@
 import streamlit as st
 from transformers import pipeline
 
-
 def main():
-    # 使用情感分析任务默认模型
-    sentiment_pipeline = pipeline("sentiment-analysis")
+    sentiment_pipeline = pipeline(model="KeviniveK/CustomModel_IMDB")
 
     st.title("Sentiment Analysis with HuggingFace Spaces")
     st.write("Enter a sentence to analyze its sentiment:")
@@ -19,4 +17,4 @@ def main():
         st.write(f"Confidence: {confidence:.2f}")
 
 if __name__ == "__main__":
-    main()
+    main()