Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ API_TOKEN = os.getenv("HF_READ_TOKEN")
|
|
15 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
16 |
timeout = 100
|
17 |
|
18 |
-
def query(prompt, is_negative=False, steps=
|
19 |
if prompt == "" or prompt == None:
|
20 |
return None
|
21 |
|
@@ -39,7 +39,7 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
|
|
39 |
"strength": strength
|
40 |
}
|
41 |
|
42 |
-
response = requests.post(
|
43 |
if response.status_code != 200:
|
44 |
print(f"Error: Failed to get image. Response status: {response.status_code}")
|
45 |
print(f"Response content: {response.text}")
|
|
|
15 |
headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
16 |
timeout = 100
|
17 |
|
18 |
+
def query(prompt, is_negative=False, steps=15, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
|
19 |
if prompt == "" or prompt == None:
|
20 |
return None
|
21 |
|
|
|
39 |
"strength": strength
|
40 |
}
|
41 |
|
42 |
+
response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
|
43 |
if response.status_code != 200:
|
44 |
print(f"Error: Failed to get image. Response status: {response.status_code}")
|
45 |
print(f"Response content: {response.text}")
|