removed input
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -40,9 +40,9 @@ class EndpointHandler():
|
|
40 |
# run inference pipeline
|
41 |
with autocast(device.type):
|
42 |
if negative_prompt is None:
|
43 |
-
image = self.pipe(
|
44 |
else:
|
45 |
-
image = self.pipe(
|
46 |
|
47 |
# encode image as base 64
|
48 |
buffered = BytesIO()
|
|
|
40 |
# run inference pipeline
|
41 |
with autocast(device.type):
|
42 |
if negative_prompt is None:
|
43 |
+
image = self.pipe(prompt = postive_prompt ,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
|
44 |
else:
|
45 |
+
image = self.pipe(prompt = postive_prompt ,negative_prompt = negative_prompt,height = height ,width = width ,guidance_scale=float(guidance_scale))["sample"][0]
|
46 |
|
47 |
# encode image as base 64
|
48 |
buffered = BytesIO()
|