civitaiarchive commited on
Commit
5291fa8
·
1 Parent(s): 85b33d8

Add civitai-to-hf-uploader

Browse files
Files changed (2) hide show
  1. app.py +12 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -8,6 +8,14 @@ import re
8
  import uuid
9
  from typing import Optional
10
  import json
 
 
 
 
 
 
 
 
11
  def download_file(url, file_path, folder):
12
  headers = {}
13
  try:
@@ -252,5 +260,9 @@ Once uploaded, it will add this repository to CivitaiArchive.com as a mirror.
252
  outputs=[output]
253
  )
254
 
 
 
 
 
255
  demo.queue(default_concurrency_limit=50)
256
  demo.launch()
 
8
  import uuid
9
  from typing import Optional
10
  import json
11
+
12
+ from apscheduler.schedulers.background import BackgroundScheduler
13
+ from huggingface_hub import Repository, HfApi
14
+ api = HfApi()
15
+ def restart_space():
16
+ api.restart_space(repo_id="civitaiarchive/civitai-to-hf-uploader", token=os.environ["HF_TOKEN"])
17
+
18
+
19
  def download_file(url, file_path, folder):
20
  headers = {}
21
  try:
 
260
  outputs=[output]
261
  )
262
 
263
+ scheduler = BackgroundScheduler()
264
+ scheduler.add_job(restart_space, 'interval', seconds=3600)
265
+ scheduler.start()
266
+
267
  demo.queue(default_concurrency_limit=50)
268
  demo.launch()
requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ APScheduler==3.10.1