hunt local

Client/hunt.py

@@ -4,6 +4,12 @@ import requests
 import io
 import zipfile
 
+local = False
+if local:
+    endpoint = "http://127.0.0.1:7860"
+else:
+    endpoint = "https://q6-p.hf.space"
+
 img_base = 'https://i.pximg.net/img-original/img/'
 
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
@@ -42,7 +48,7 @@ for index in indexs:
 
         print(f"Group: {group_name}\nFiltered: {len(filtered)}/{len(post_ids)}")
         if filtered:
-            data = requests.post('https://q6-p.hf.space/pixif', json={"post_ids": filtered}).json()
+            data = requests.post(f'{endpoint}/pixif', json={"post_ids": filtered}).json()
 
             for post_id, url in data.items():
                 txn.put(post_id.encode(), url.encode())
@@ -56,7 +62,7 @@ for index in indexs:
     to_download = {post_id: url for post_id, url in post_ids.items() if url not in blacklist and f"{post_id}.png" not in images_cache}
     print(f"Downloading {len(to_download)} images...")
     if to_download:
-        images_zip = requests.post('https://q6-p.hf.space/download', json={'posts': to_download}, stream=True)
+        images_zip = requests.post(f'{endpoint}/download', json={'posts': to_download}, stream=True)
 
         print("Extracting images...")
         with io.BytesIO(images_zip.content) as f: