Spaces:

themeetjani
/

tweet_classification

Sleeping

themeetjani commited on Aug 12, 2023

Commit

43451e6

•

1 Parent(s): 1ebcf26

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ import pandas as pd
 from transformers import BertTokenizer, BertModel
 from torch import cuda
 import gradio as gr
-#device = 'cuda' if cuda.is_available() else 'cpu'
 class RobertaClass(torch.nn.Module):
     def __init__(self):

 from transformers import BertTokenizer, BertModel
 from torch import cuda
 import gradio as gr
+device = 'cuda' if cuda.is_available() else 'cpu'
 class RobertaClass(torch.nn.Module):
     def __init__(self):