khurrameycon commited on
Commit
98a4f20
·
verified ·
1 Parent(s): 57cdf69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -84,8 +84,17 @@ def predict_image(image_url, text, file_pref):
84
  response.raise_for_status()
85
  image = Image.open(io.BytesIO(response.content)).convert("RGB")
86
  else:
 
87
  image = extract_image_from_pdf(image_url)
88
 
 
 
 
 
 
 
 
 
89
  streamer = TextStreamer()
90
 
91
  def generate_response():
 
84
  response.raise_for_status()
85
  image = Image.open(io.BytesIO(response.content)).convert("RGB")
86
  else:
87
+ print("Extracting image from PDF...")
88
  image = extract_image_from_pdf(image_url)
89
 
90
+ # Convert the image to base64
91
+ buffered = io.BytesIO()
92
+ image.save(buffered, format="PNG")
93
+ img_str = base64.b64encode(buffered.getvalue()).decode()
94
+ image_url = f"data:image/png;base64,{img_str}"
95
+
96
+ print(f"Image URL being sent to Groq: {image_url[:100]}...")
97
+
98
  streamer = TextStreamer()
99
 
100
  def generate_response():