Joshua1808 commited on
Commit
336e489
·
1 Parent(s): 07f4cd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -5,7 +5,23 @@ import torch
5
  import numpy as np
6
  import regex as re
7
 
 
8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  consumer_key = "BjipwQslVG4vBdy4qK318KnoA"
10
  consumer_secret = "3fzL70v9faklrPgvTi3zbofw9rwk92fgGdtAslFkFYt8kGmqBJ"
11
  access_token = "1217853705086799872-Y5zEChpTeKccuLY3XJRXDPPZhNrlba"
 
5
  import numpy as np
6
  import regex as re
7
 
8
+ from pysentimiento.preprocessing import preprocess_tweet
9
 
10
+ from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
11
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification,AdamW
12
+ tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/twitter_sexismo-finetuned-robertuito-exist2021')
13
+ model = AutoModelForSequenceClassification.from_pretrained("hackathon-pln-es/twitter_sexismo-finetuned-robertuito-exist2021")
14
+
15
+ import torch
16
+ if torch.cuda.is_available():
17
+ device = torch.device( "cuda")
18
+ print('I will use the GPU:', torch.cuda.get_device_name(0))
19
+
20
+ else:
21
+ print('No GPU available, using the CPU instead.')
22
+ device = torch.device("cpu")
23
+
24
+
25
  consumer_key = "BjipwQslVG4vBdy4qK318KnoA"
26
  consumer_secret = "3fzL70v9faklrPgvTi3zbofw9rwk92fgGdtAslFkFYt8kGmqBJ"
27
  access_token = "1217853705086799872-Y5zEChpTeKccuLY3XJRXDPPZhNrlba"