Mridul2003 commited on
Commit
2fb729d
·
verified ·
1 Parent(s): 31aab23

Update services/image_ocr.py

Browse files
Files changed (1) hide show
  1. services/image_ocr.py +2 -2
services/image_ocr.py CHANGED
@@ -11,10 +11,10 @@ class ImageOCRService:
11
  response = requests.get(image_url)
12
  image = Image.open(BytesIO(response.content)).convert("RGB")
13
 
14
- result = self.pipe(image, text=[{
15
  "role": "user",
16
  "content": [
17
- {"type": "image", "url": image_url},
18
  {"type": "text", "text": "extract text from image"}
19
  ]
20
  }])
 
11
  response = requests.get(image_url)
12
  image = Image.open(BytesIO(response.content)).convert("RGB")
13
 
14
+ result = self.pipe([{
15
  "role": "user",
16
  "content": [
17
+ {"type": "image", "image": image},
18
  {"type": "text", "text": "extract text from image"}
19
  ]
20
  }])