amrtweg commited on
Commit
a2c85d2
·
verified ·
1 Parent(s): 3de6e6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,8 +2,9 @@ import gradio as gr
2
  import torch
3
  from actora import ActoraPredictor, emoji_map as actora_emoji_map
4
  import warnings
5
- warnings.filterwarnings("ignore")
6
-
 
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
  repo_id = "amrtweg/Actora"
 
2
  import torch
3
  from actora import ActoraPredictor, emoji_map as actora_emoji_map
4
  import warnings
5
+ warnings.filterwarnings("ignore", category=UserWarning)
6
+ import logging
7
+ logging.getLogger("transformers").setLevel(logging.ERROR)
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  repo_id = "amrtweg/Actora"