kozy9 commited on
Commit
36c47de
·
verified ·
1 Parent(s): 888adc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -3,7 +3,6 @@ from transformers import pipeline
3
 
4
  news_classifier = pipeline(task="text-classification",
5
  model='kozy9/news_classification_file',
6
- device=torch.device("cuda") if torch.cuda.is_available() else "cpu",
7
  top_k=1,
8
  batch_size=32)
9
 
 
3
 
4
  news_classifier = pipeline(task="text-classification",
5
  model='kozy9/news_classification_file',
 
6
  top_k=1,
7
  batch_size=32)
8