Iker commited on
Commit
7ddc380
1 Parent(s): 2d8250d
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -6,14 +6,16 @@ from cache_system import CacheHandler
6
  from gradio_client import Client
7
  from collections import OrderedDict
8
  from typing import Any
 
9
 
10
- print(f"CPU cores: {os.cpu_count()}.")
11
 
12
  server = os.environ.get("SERVER") or True
13
  auth_token = os.environ.get("TOKEN") or True
14
 
15
  client = Client(server, hf_token=auth_token, verbose=False)
16
 
 
 
17
 
18
  class HuggingFaceDatasetSaver_custom(gr.HuggingFaceDatasetSaver):
19
  def _deserialize_components(
@@ -53,6 +55,10 @@ def generate_text(
53
  url: str, mode: int, progress=gr.Progress(track_tqdm=False)
54
  ) -> (str, str):
55
  global cache_handler
 
 
 
 
56
 
57
  url = url.strip()
58
 
 
6
  from gradio_client import Client
7
  from collections import OrderedDict
8
  from typing import Any
9
+ import datetime
10
 
 
11
 
12
  server = os.environ.get("SERVER") or True
13
  auth_token = os.environ.get("TOKEN") or True
14
 
15
  client = Client(server, hf_token=auth_token, verbose=False)
16
 
17
+ total_runs = 0
18
+
19
 
20
  class HuggingFaceDatasetSaver_custom(gr.HuggingFaceDatasetSaver):
21
  def _deserialize_components(
 
55
  url: str, mode: int, progress=gr.Progress(track_tqdm=False)
56
  ) -> (str, str):
57
  global cache_handler
58
+ global total_runs
59
+
60
+ total_runs += 1
61
+ print(f"Total runs: {total_runs}. Last run: {datetime.datetime.now()}")
62
 
63
  url = url.strip()
64