Spaces:

CatoEr
/

Profiling-Social-Media-Users

Sleeping

CatoEr commited on May 30

Commit

6e257b4

•

1 Parent(s): bea74aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,11 +3,8 @@ from torch import nn
 from transformers import AutoModel, AutoTokenizer
 import gradio as gr
-# Check if CUDA is available
-if torch.cuda.is_available():
-    device = torch.device("cuda")
-else:
-    device = torch.device("cpu")
 class RaceClassifier(nn.Module):

 from transformers import AutoModel, AutoTokenizer
 import gradio as gr
+device = torch.device("cpu")
 class RaceClassifier(nn.Module):