Update main.py

main.py

@@ -64,17 +64,22 @@ def send_it1(inputs, noise_level, proc=proc1):
     output = proc(prompt_with_noise)
     return output
     
-def generate_image(inputs,is_negative, steps, cfg_scale, seed):
+def generate_image(inputs, is_negative, steps, cfg_scale, seed):
     try:
         global request_counter
         request_counter += 1
         timestamp = f"{time.time()}_{request_counter}"
         prompt_with_noise = add_random_noise(inputs) + f" - {timestamp}"
-        payload = {"inputs": prompt_with_noise,
-                   "is_negative": is_negative,
-                   "steps": steps,
-                   "cfg_scale": cfg_scale,
-                   "seed": seed if seed is not None else random.randint(-1, 2147483647) }
+        image_bytes = query({
+            "inputs": "Astronaut riding a horse",
+        })
+        payload = {
+            "inputs": prompt_with_noise,
+            "is_negative": is_negative,
+            "steps": steps,
+            "cfg_scale": cfg_scale,
+            "seed": seed if seed is not None else random.randint(-1, 2147483647)
+        }
         response = requests.post(API_URL, headers=headers, json=payload)
         response.raise_for_status()  # Raise an exception for HTTP errors
         image_bytes = response.content