neuralleap commited on
Commit
46a1568
·
verified ·
1 Parent(s): c1ba1a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -33,13 +33,15 @@ Extract the following fields as JSON:
33
  def process_image(image: Image.Image):
34
  buffered = io.BytesIO()
35
  image.save(buffered, format="JPEG")
36
- base64_image = base64.b64encode(buffered.getvalue()).decode()
37
 
38
- contents = [
39
- genai.Content.create(role="user", parts=[genai.Part.from_text(prompt)]),
40
- genai.Content.create(role="user", parts=[genai.Part.from_data(base64.b64decode(base64_image), mime_type="image/jpeg")])
41
- ]
42
- response = model.generate_content(contents)
 
 
43
  return response.text
44
 
45
  # Gradio interface
@@ -47,7 +49,7 @@ demo = gr.Interface(
47
  fn=process_image,
48
  inputs=gr.Image(type="pil"),
49
  outputs="textbox",
50
- title="Healthelic Form Data Extractor (Doc Scaner)",
51
  description="Upload a scanned medical form to extract key fields."
52
  )
53
 
 
33
  def process_image(image: Image.Image):
34
  buffered = io.BytesIO()
35
  image.save(buffered, format="JPEG")
36
+ base64_image = buffered.getvalue()
37
 
38
+ response = model.generate_content([
39
+ prompt,
40
+ {
41
+ "mime_type": "image/jpeg",
42
+ "data": base64_image
43
+ }
44
+ ])
45
  return response.text
46
 
47
  # Gradio interface
 
49
  fn=process_image,
50
  inputs=gr.Image(type="pil"),
51
  outputs="textbox",
52
+ title="Healthelic Form Data Extractor (Doc Scaner) - Gemini",
53
  description="Upload a scanned medical form to extract key fields."
54
  )
55