Update app.py

app.py

@@ -280,8 +280,10 @@ def generate_query(formatted_predictions):
       What features of the chest X-ray image support the predicted condition(s)?
       """
 
-def predictionReportGenerator(vlm_model, image_path, system_prompt, query_prompt):
-    image_ = Image.open(image_path).convert("RGB")
+def predictionReportGenerator(vlm_model, img_pil, system_prompt, query_prompt):
+    # image_ = Image.open(image_path).convert("RGB")
+    image_ = img_pil.convert("RGB")
+
     messages = [
     {
         "role": "system",
@@ -351,7 +353,7 @@ def run_diagnosis(
         formatted_predictions = {label: output_probs[label] for label in predicted_classes}
         
         query_prompt = generate_query(formatted_predictions)
-        explanation_ = predictionReportGenerator(vlm_model = VLM_model, image_path = input_image, 
+        explanation_ = predictionReportGenerator(vlm_model = VLM_model, img_pil = input_image, 
                                                  system_prompt = VLM_SYSTEM_PROMPT, query_prompt = query_prompt)
 
     return "\n".join(predicted_classes), explanation_, output_probs