Skip to content

Commit

Permalink
print response
Browse files Browse the repository at this point in the history
  • Loading branch information
kushal-10 committed Apr 29, 2024
1 parent bbd06f8 commit f659d61
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions backends/huggingface_multimodal_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,8 @@ def generate_response(self, messages: List[Dict]) -> Tuple[Any, Any, str]:
# Store generated text
response = {'response': generated_text}

print(f"Response: {response}\n")

response_text = generated_text[0].split(self.cull)[-1] # Get the last assistant response

return prompt, response, response_text

0 comments on commit f659d61

Please sign in to comment.