Luciferalive commited on
Commit
11c95b0
1 Parent(s): 038b623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from transformers import pipeline
4
 
5
  # Load the model
6
  model_name = "knowledgator/comprehend_it-base"
7
- classifier = pipeline("zero-shot-classification", model=model_name, device=0)
 
8
 
9
  # Function to classify feedback
10
  def classify_feedback(feedback_text):
 
4
 
5
  # Load the model
6
  model_name = "knowledgator/comprehend_it-base"
7
+ classifier = pipeline("zero-shot-classification", model=model_name, device="cpu")
8
+
9
 
10
  # Function to classify feedback
11
  def classify_feedback(feedback_text):