saattrupdan commited on
Commit
90ed2ce
·
1 Parent(s): c21078f

fix: Type hints

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -7,6 +7,7 @@ import gradio as gr
7
  from transformers import pipeline
8
  from shap import Explainer
9
  import numpy as np
 
10
 
11
 
12
  def main():
@@ -27,8 +28,8 @@ def main():
27
  "😊",
28
  ]
29
 
30
- def classification(text) -> tuple[dict[str, float], dict]:
31
- output: list[dict] = pipe(text)[0]
32
  print(output)
33
 
34
  explainer = Explainer(pipe)
@@ -40,7 +41,7 @@ def main():
40
  if np.abs(shap_values).max() <= boundary:
41
  boundary = np.abs(shap_values).max() - 1e-6
42
 
43
- words: list[str] = explanation.data[0]
44
  records = list()
45
  char_idx = 0
46
  for word, shap_value in zip(words, shap_values):
 
7
  from transformers import pipeline
8
  from shap import Explainer
9
  import numpy as np
10
+ from typing import Tuple, Dict, List
11
 
12
 
13
  def main():
 
28
  "😊",
29
  ]
30
 
31
+ def classification(text) -> Tuple[Dict[str, float], dict]:
32
+ output: List[dict] = pipe(text)[0]
33
  print(output)
34
 
35
  explainer = Explainer(pipe)
 
41
  if np.abs(shap_values).max() <= boundary:
42
  boundary = np.abs(shap_values).max() - 1e-6
43
 
44
+ words: List[str] = explanation.data[0]
45
  records = list()
46
  char_idx = 0
47
  for word, shap_value in zip(words, shap_values):