add image conversion to jpeg

main.py

@@ -33,9 +33,14 @@ decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, r
 
 def generateOutput(fileData):
     pil_image = Image.open(BytesIO(fileData))
-    pil_image.resize((800, 600))
-    pixel_values = processor(pil_image, return_tensors="pt").pixel_values
-
+    resized_image = pil_image.resize((800, 600)).convert('RGB')
+    rgb_image = Image.new('RGB', resized_image.size)
+    rgb_image.paste(resized_image)
+    output_buffer = BytesIO()
+    rgb_image.save(output_buffer, format="JPEG", quality = 100)
+    jpeg_image = Image.open(BytesIO(output_buffer.getvalue()))
+    
+    pixel_values = processor(jpeg_image, return_tensors="pt").pixel_values
     outputs = model.generate(
         pixel_values.to(device),
         decoder_input_ids=decoder_input_ids.to(device),