Commit
·
f8291d9
1
Parent(s):
dfd06c6
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,19 +14,11 @@ class Prodia:
|
|
| 14 |
self.headers = {
|
| 15 |
"X-Prodia-Key": api_key
|
| 16 |
}
|
| 17 |
-
|
| 18 |
def generate(self, params):
|
| 19 |
response = self._post(f"{self.base}/sdxl/generate", params)
|
| 20 |
return response.json()
|
| 21 |
-
|
| 22 |
-
def transform(self, params):
|
| 23 |
-
response = self._post(f"{self.base}/sd/transform", params)
|
| 24 |
-
return response.json()
|
| 25 |
-
|
| 26 |
-
def controlnet(self, params):
|
| 27 |
-
response = self._post(f"{self.base}/sd/controlnet", params)
|
| 28 |
-
return response.json()
|
| 29 |
-
|
| 30 |
def get_job(self, job_id):
|
| 31 |
response = self._get(f"{self.base}/job/{job_id}")
|
| 32 |
return response.json()
|
|
@@ -71,7 +63,7 @@ def image_to_base64(image_path):
|
|
| 71 |
# Convert the image to bytes
|
| 72 |
buffered = BytesIO()
|
| 73 |
image.save(buffered, format="PNG") # You can change format to PNG if needed
|
| 74 |
-
|
| 75 |
# Encode the bytes to base64
|
| 76 |
img_str = base64.b64encode(buffered.getvalue())
|
| 77 |
|
|
|
|
| 14 |
self.headers = {
|
| 15 |
"X-Prodia-Key": api_key
|
| 16 |
}
|
| 17 |
+
|
| 18 |
def generate(self, params):
|
| 19 |
response = self._post(f"{self.base}/sdxl/generate", params)
|
| 20 |
return response.json()
|
| 21 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
def get_job(self, job_id):
|
| 23 |
response = self._get(f"{self.base}/job/{job_id}")
|
| 24 |
return response.json()
|
|
|
|
| 63 |
# Convert the image to bytes
|
| 64 |
buffered = BytesIO()
|
| 65 |
image.save(buffered, format="PNG") # You can change format to PNG if needed
|
| 66 |
+
|
| 67 |
# Encode the bytes to base64
|
| 68 |
img_str = base64.b64encode(buffered.getvalue())
|
| 69 |
|