diff --git a/core/views.py b/core/views.py index bc25418..68676c9 100644 --- a/core/views.py +++ b/core/views.py @@ -5,7 +5,6 @@ from django.contrib import messages import requests import json -from transformers import pipeline from .forms import RegisterForm from .models import User, UserInteraction @@ -440,6 +439,7 @@ def analyze_sentiment(feeling_text): global sentiment_pipeline try: if sentiment_pipeline is None: + from transformers import pipeline sentiment_pipeline = pipeline("sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") result = sentiment_pipeline(feeling_text)[0] return result['label'], result['score'] diff --git a/requirements.txt b/requirements.txt index 1ce4e9b..08e2d14 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ Django Torch transformers -pytz \ No newline at end of file +pytz +requests \ No newline at end of file