local users

Client/Extract Pixiv/ai_search.py

@@ -5,7 +5,7 @@ local = False
 if local:
     endpoint = "http://127.0.0.1:7860"
 else:
-    endpoint = "https://q6-p.hf.space/"
+    endpoint = "https://q6-p.hf.space"
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
 input_url = input("Enter the URL: ")

Client/Extract Pixiv/user.py

@@ -2,6 +2,12 @@ import requests
 import re
 import os
 
+local = False
+if local:
+    endpoint = "http://127.0.0.1:7860"
+else:
+    endpoint = "https://q6-p.hf.space"
+
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 
 inp = input("Enter User IDs (separated by spaces or commas): ")
@@ -9,7 +15,7 @@ user_ids = re.findall(r"\d+", inp)
 
 user_ids = [int(uid) for uid in user_ids]
 
-response = requests.get('https://q6-p.hf.space/users', params={'user_ids': user_ids})
+response = requests.get(f'{endpoint}/users', params={'user_ids': user_ids})
 response.raise_for_status()
 data = response.json()
 

run_docker.bat

@@ -0,0 +1 @@
+docker run -p 7860:7860 pixif