Xylor commited on
Commit
131144c
·
verified ·
1 Parent(s): a924ecb

Added logging

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -1,6 +1,12 @@
1
  import gradio
2
-
 
3
  from transformers import pipeline
 
 
 
 
 
4
  classifier = pipeline("zero-shot-classification",
5
  model="facebook/bart-large-mnli")
6
 
@@ -27,7 +33,12 @@ def transform_output(res: dict) -> list:
27
  )
28
 
29
  def clf_text(txt: str | list[str]):
 
 
30
  res = classifier(txt, categories, multi_label=True)
 
 
 
31
  if isinstance(res, list):
32
  return [ transform_output(dct) for dct in res ]
33
  else:
@@ -52,4 +63,5 @@ gradio_interface = gradio.Interface(
52
  inputs = "text",
53
  outputs = gradio.JSON()
54
  )
 
55
  gradio_interface.launch()
 
1
  import gradio
2
+ import logging
3
+ import time
4
  from transformers import pipeline
5
+
6
+ logger = logging.getLogger()
7
+ logger.setLevel(logging.DEBUG)
8
+ logging.debug("Starting logging for gradio_test_001.")
9
+
10
  classifier = pipeline("zero-shot-classification",
11
  model="facebook/bart-large-mnli")
12
 
 
33
  )
34
 
35
  def clf_text(txt: str | list[str]):
36
+ logger.debug("Classify: " + repr(txt))
37
+ t0 = time.time()
38
  res = classifier(txt, categories, multi_label=True)
39
+ elapsed = time.time() - t0
40
+ logger.debug(f"Done. {elapsed:.02f}s")
41
+ logger.debug(f"Result(s):", res)
42
  if isinstance(res, list):
43
  return [ transform_output(dct) for dct in res ]
44
  else:
 
63
  inputs = "text",
64
  outputs = gradio.JSON()
65
  )
66
+ logger.debug("Launch app.")
67
  gradio_interface.launch()