Hugging Face's logo

Spaces:
CatoEr
/
Profiling-Social-Media-Users
Sleeping

App Files Community
CatoEr commited on
Commit
6e257b4
·
verified ·
1 Parent(s): bea74aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -3,11 +3,8 @@ from torch import nn
3
  from transformers import AutoModel, AutoTokenizer
4
  import gradio as gr
5
 
6
- # Check if CUDA is available
7
- if torch.cuda.is_available():
8
- device = torch.device("cuda")
9
- else:
10
- device = torch.device("cpu")
11
 
12
 
13
  class RaceClassifier(nn.Module):
 
3
  from transformers import AutoModel, AutoTokenizer
4
  import gradio as gr
5
 
6
+
7
+ device = torch.device("cpu")
 
 
 
8
 
9
 
10
  class RaceClassifier(nn.Module):