Kseniia-Kholina commited on
Commit
3f8bb76
·
verified ·
1 Parent(s): 968e490

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -3,6 +3,11 @@ import pandas as pd
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForMaskedLM
5
  import logging
 
 
 
 
 
6
 
7
  logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForMaskedLM
5
  import logging
6
+ import numpy as np
7
+ import matplotlib.pyplot as plt
8
+ import seaborn as sns
9
+ from io import BytesIO
10
+ from PIL import Image
11
 
12
  logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")