Commit
·
6140afd
1
Parent(s):
61b80cc
Upload handler.py
Browse files- handler.py +0 -1
handler.py
CHANGED
|
@@ -52,7 +52,6 @@ class EndpointHandler():
|
|
| 52 |
generated_ids = self.generate_model.generate(**prompt_image_processed)
|
| 53 |
generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
|
| 54 |
result["image_prompt"] = generated_text
|
| 55 |
-
pass
|
| 56 |
if extract_feature:
|
| 57 |
prompt_feature = self.feature_model(**prompt_image_processed)
|
| 58 |
result["prompt_feature"] = prompt_feature
|
|
|
|
| 52 |
generated_ids = self.generate_model.generate(**prompt_image_processed)
|
| 53 |
generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
|
| 54 |
result["image_prompt"] = generated_text
|
|
|
|
| 55 |
if extract_feature:
|
| 56 |
prompt_feature = self.feature_model(**prompt_image_processed)
|
| 57 |
result["prompt_feature"] = prompt_feature
|