multimodalart HF staff commited on
Commit
a5e7022
1 Parent(s): 1983d3d

Update handler.py

Browse files
Files changed (1) hide show
  1. 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
- full_prompt,
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