Spaces:
Runtime error
Runtime error
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}")
|