Spaces:

cwchang
/

text_classification_example

Sleeping

cwchang commited on Dec 1, 2023

Commit

cb66b8a

1 Parent(s): 351c92c

use cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-classifier = pipeline(task='text-classification', model='cwchang/my-text-classification-finetuned-v1', device=0)
 def classify(text):
     return classifier(text)[0]["label"]

 import gradio as gr
 from transformers import pipeline
+classifier = pipeline(task='text-classification', model='cwchang/my-text-classification-finetuned-v1', device=-1)
 def classify(text):
     return classifier(text)[0]["label"]