Diego Carpintero commited on
Commit
d37f373
1 Parent(s): c64e302

refactor analyze method

Browse files
Files changed (1) hide show
  1. minerva.py +2 -3
minerva.py CHANGED
@@ -117,12 +117,11 @@ class Minerva:
117
  """Reset team state"""
118
  self.team.reset()
119
 
120
- async def analyze_image(self, image_path: str) -> AsyncIterator:
121
  """
122
  Analyze an image for potential scams.
123
  """
124
- pil_image = Image.open(image_path)
125
- img = AGImage(pil_image)
126
  mm_message = MultiModalMessage(content=[img], source="User")
127
 
128
  return self.team.run_stream(task=mm_message)
 
117
  """Reset team state"""
118
  self.team.reset()
119
 
120
+ async def analyze(self, image: Image) -> AsyncIterator:
121
  """
122
  Analyze an image for potential scams.
123
  """
124
+ img = AGImage(image)
 
125
  mm_message = MultiModalMessage(content=[img], source="User")
126
 
127
  return self.team.run_stream(task=mm_message)