Walmart-the-bag commited on
Commit
2fc1e78
1 Parent(s): b049ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -6,7 +6,6 @@ import time
6
  url = "https://nexra.aryahcr.cc/api/image/complements"
7
  headers = {"Content-Type": "application/json"}
8
 
9
- no_nsfw_image_path = "no_nsfw.png"
10
 
11
  def generate_image(prompt, model):
12
  data = {"prompt": prompt, "model": model}
@@ -15,9 +14,6 @@ def generate_image(prompt, model):
15
  if response.status_code == 200:
16
  response_data = response.json()
17
  image_id = response_data.get("id")
18
-
19
- if not image_id:
20
- return no_nsfw_image_path
21
 
22
  while True:
23
  status_response = requests.get(f"{url}/{image_id}")
 
6
  url = "https://nexra.aryahcr.cc/api/image/complements"
7
  headers = {"Content-Type": "application/json"}
8
 
 
9
 
10
  def generate_image(prompt, model):
11
  data = {"prompt": prompt, "model": model}
 
14
  if response.status_code == 200:
15
  response_data = response.json()
16
  image_id = response_data.get("id")
 
 
 
17
 
18
  while True:
19
  status_response = requests.get(f"{url}/{image_id}")