Hugging Face's logo

Spaces:
abdulmatinomotoso
/
Github_issues_search_engine
Runtime error

App Files Community
abdulmatinomotoso commited on
Commit
2b06616
·
1 Parent(s): 3bd30c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def concatenate_text(examples):
53
 
54
  comments_dataset = comments_dataset.map(concatenate_text)
55
 
56
- device = torch.device("cuda")
57
  model = model.to(device)
58
 
59
  def cls_pooling(model_output):
 
53
 
54
  comments_dataset = comments_dataset.map(concatenate_text)
55
 
56
+ device = torch.device("cpu")
57
  model = model.to(device)
58
 
59
  def cls_pooling(model_output):