alvarobartt
/

SmolVLM-Instruct-Handler

Image-Text-to-Text

Model card Files Files and versions

alvarobartt HF Staff commited on Dec 4, 2024

Commit

d1cf6f0

·

verified ·

1 Parent(s): 89baa6d

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -100,6 +100,6 @@ class EndpointHandler:
                 generated_ids,
                 skip_special_tokens=True,
             )
-            predictions.append(generated_texts[0][len(prompt):])
         return {"predictions": predictions}

                 generated_ids,
                 skip_special_tokens=True,
             )
+            predictions.append(generated_texts[0])
         return {"predictions": predictions}