Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -74,8 +74,8 @@ class EndpointHandler:
|
|
| 74 |
output = self.pipe(
|
| 75 |
prompt=prompt,
|
| 76 |
image=image,
|
| 77 |
-
num_inference_steps=
|
| 78 |
-
guidance_scale=
|
| 79 |
).images[0]
|
| 80 |
print("🎨 Image generated.")
|
| 81 |
except Exception as e:
|
|
|
|
| 74 |
output = self.pipe(
|
| 75 |
prompt=prompt,
|
| 76 |
image=image,
|
| 77 |
+
num_inference_steps=20,
|
| 78 |
+
guidance_scale=2.0
|
| 79 |
).images[0]
|
| 80 |
print("🎨 Image generated.")
|
| 81 |
except Exception as e:
|