Spaces:

zhouxiaoxi
/

text-sentiment-analysis

Runtime error

zhouxiaoxi commited on Aug 5

Commit

0bd42e9

verified ·

1 Parent(s): ae067e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,8 @@ import torch
 import torch.nn as nn
 from transformers import BertTokenizer, BertModel
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 names = ['负向', '正向']
 # 分词器

 import torch.nn as nn
 from transformers import BertTokenizer, BertModel
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = "cpu"
 names = ['负向', '正向']
 # 分词器