Spaces:

Akshayram1
/

personalize_music_app

Sleeping

Akshayram1 commited on Jan 8

Commit

0f1be6d

verified ·

1 Parent(s): 2bb9b0c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import pipeline
 import google.generativeai as genai
 from pytube import Search
-# Load sentiment analysis model
-mood_classifier = pipeline("sentiment-analysis")
 # Functions
 def detect_mood(text):

 import google.generativeai as genai
 from pytube import Search
+# Load sentiment analysis model using PyTorch backend
+mood_classifier = pipeline("sentiment-analysis", framework="pt")  # Use PyTorch
 # Functions
 def detect_mood(text):