Update handler.py
Browse files- handler.py +1 -2
handler.py
CHANGED
@@ -44,9 +44,8 @@ class EndpointHandler:
|
|
44 |
inputs = data.pop("inputs", data)
|
45 |
|
46 |
# Automatically add trigger tokens to the beginning of the prompt
|
47 |
-
full_prompt = f"A <s0><s1> emoji {inputs}"
|
48 |
images = self.pipe(
|
49 |
-
|
50 |
cross_attention_kwargs={"scale": 0.8},
|
51 |
num_inference_steps=25
|
52 |
).images
|
|
|
44 |
inputs = data.pop("inputs", data)
|
45 |
|
46 |
# Automatically add trigger tokens to the beginning of the prompt
|
|
|
47 |
images = self.pipe(
|
48 |
+
inputs,
|
49 |
cross_attention_kwargs={"scale": 0.8},
|
50 |
num_inference_steps=25
|
51 |
).images
|