Lord-Raven commited on
Commit
6d6ef9c
·
1 Parent(s): 89be057

Turning on logging to diagnose another issue.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def classify(data_string, request: gradio.Request):
49
  try:
50
  if 'cpu' not in data:
51
  result = zero_shot_classification_gpu(data)
52
- print(f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - GPU Classification took {time.time() - start_time}")
53
  except Exception as e:
54
  print(f"GPU classification failed: {e}\nFall back to CPU.")
55
  if not result:
 
49
  try:
50
  if 'cpu' not in data:
51
  result = zero_shot_classification_gpu(data)
52
+ print(f"{datetime.now().strftime('%Y-%m-%d %H:%M:%S')} - GPU Classification took {time.time() - start_time}: {data}")
53
  except Exception as e:
54
  print(f"GPU classification failed: {e}\nFall back to CPU.")
55
  if not result: