Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from fastai.vision.all import *
 from PIL import Image
+import io
+import numpy as np
 
 
 learn = load_learner('model.pkl')
@@ -8,7 +10,7 @@ labels = learn.dls.vocab
 input_shape = (512, 512)  # Dimensões desejadas para a entrada
 
 def preprocess_image(img):
-    img = Image.open(img)
+    img = Image.fromarray(np.uint8(img))
     img = img.resize(input_shape)
     return img
 
@@ -20,7 +22,7 @@ def predict(img):
 
 title = "Tuberculosis detector"
 description = "A tuberculosis detector trained on 3700 chest X-ray images. Created as a demo for Gradio and HuggingFace Spaces."
-examples = ['1.jpg', '2.jpeg', '3.jpg', '4.jpg']
+examples = ['1.jpeg', '2.jpeg', '3.jpeg', '4.jpeg']
 interpretation = 'default'
 enable_queue = True