luisresende13
/

llava-interleave-qwen-0.5b-hf

Image-Text-to-Text

Model card Files Files and versions

luisresende13 commited on Jul 21, 2024

Commit

080a129

·

verified ·

1 Parent(s): c828564

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,5 +26,5 @@ class EndpointHandler():
         results = self.pipe(image, prompt=prompt, generate_kwargs={"max_new_tokens": max_new_tokens})
         result = results[0]
-        # result['generated_text'] = result['generated_text'].replace(prompt.replace('<image>', ''), '')
         return result

         results = self.pipe(image, prompt=prompt, generate_kwargs={"max_new_tokens": max_new_tokens})
         result = results[0]
+        result['generated_text'] = result['generated_text'].replace(prompt.replace('<image>', '') + ' ', '')
         return result