SineTheta commited on
Commit
6dad16b
·
1 Parent(s): c34d1aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,14 +1,17 @@
1
  import gradio as gr
2
  import re
 
3
  import subprocess
4
  import sys
5
 
6
- def install(package):
7
- subprocess.check_call([sys.executable, "-m", "pip", "install", package])
8
-
9
- install(transformers)
 
 
10
 
11
- from transformers import pipeline
12
 
13
  sp_model ="JonatanGk/roberta-base-bne-finetuned-cyberbullying-spanish"
14
  ca_model = "JonatanGk/roberta-base-ca-finetuned-cyberbullying-catalan"
 
1
  import gradio as gr
2
  import re
3
+
4
  import subprocess
5
  import sys
6
 
7
+ try:
8
+ from transformers import pipeline
9
+ except ModuleNotFoundError:
10
+ subprocess.check_call([sys.executable, "-m", "pip", "install", 'transformers'])
11
+ finally:
12
+ from transformers import pipeline
13
 
14
+
15
 
16
  sp_model ="JonatanGk/roberta-base-bne-finetuned-cyberbullying-spanish"
17
  ca_model = "JonatanGk/roberta-base-ca-finetuned-cyberbullying-catalan"