Verify requests

src/model_demo.py

@@ -17,20 +17,21 @@ current_model: str
 last_current_model_sync: datetime = datetime.fromtimestamp(0, TIMEZONE)
 
 def get_current_model() -> str | None:
+    global current_model
+    global last_current_model_sync
+    now = datetime.now(TIMEZONE)
+    if now - last_current_model_sync < timedelta(minutes=1):
+        return current_model
+    last_current_model_sync = now
+
     try:
-        global current_model
-        global last_current_model_sync
-        now = datetime.now(TIMEZONE)
-        if now - last_current_model_sync < timedelta(minutes=5):
-            return current_model
-        last_current_model_sync = now
-
-        response = requests.get(f"{SERVER_API}/model", verify=False)
+        response = requests.get(f"{SERVER_API}/model")
         response.raise_for_status()
         model = response.json()
         current_model = f"{model['uid']} - {model['url']}"
         return current_model
     except:
+        print("Unable to connect to API")
         return None
 
 
@@ -49,12 +50,13 @@ def submit(prompt: str, seed: int | str | None, baseline: bool) -> tuple:
         seed = random.randint(0, 2 ** 32 - 1)
 
     print(f"Making request with prompt: {prompt}, seed: {seed}, baseline: {baseline}")
-    response = requests.post(f"{SERVER_API}/generate", params={"prompt": prompt, "baseline": baseline, "seed": seed}, verify=False)
+    response = requests.post(f"{SERVER_API}/generate", params={"prompt": prompt, "baseline": baseline, "seed": seed})
     response.raise_for_status()
     result = response.json()
     generation_time = float(result["generation_time"])
+    nsfw = result["nsfw"]
     image = image_from_base64(result["image"])
-    print(f"Received image with generation time: {generation_time:.3f}s")
+    print(f"Received image with generation time: {generation_time:.3f}s, NSFW: {nsfw}")
 
     return None, gr.Image(
         image,