Update app.py

app.py

@@ -35,8 +35,11 @@ class ThyroidTumorClassifierApp:
         # Add information to the output image
         output_image_with_info = self.add_info_to_image(image, predicted_label, probabilities)
         
-        # Return the modified output image as an array
-        return output_image_with_info
+        # Convert the output image to PIL format
+        output_pil_image = Image.fromarray(output_image_with_info)
+        
+        # Return the modified output image as PIL image
+        return output_pil_image
 
     def add_info_to_image(self, image, predicted_label, probabilities):
         # Convert the image to Pillow image format
@@ -68,7 +71,7 @@ class ThyroidTumorClassifierApp:
         input_interface = gr.Interface(
             fn=self.classify_image,
             inputs=gr.inputs.Image(),
-            outputs=gr.outputs.Image(),
+            outputs=gr.outputs.Image(type="pil"),  # Use type="pil" for Pillow image format
             title="Tumor da Tireoide Classificação",
             description="Faça o upload de uma imagem de um tumor da tireoide para classificação. A saída inclui o rótulo da classe prevista e as probabilidades com informações adicionais.",
         )