correcao
Browse files
app.py
CHANGED
|
@@ -11,16 +11,18 @@ interpreter.allocate_tensors()
|
|
| 11 |
input_details = interpreter.get_input_details()
|
| 12 |
output_details = interpreter.get_output_details()
|
| 13 |
|
| 14 |
-
|
| 15 |
def predict(image):
|
| 16 |
# Verifica se a imagem está no formato correto (RGB)
|
| 17 |
-
if image.
|
| 18 |
-
image
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
-
#
|
| 21 |
-
image = image.
|
| 22 |
-
image = np.
|
| 23 |
-
image = np.expand_dims(image, axis=0) # Adiciona a dimensão extra para o batch
|
| 24 |
|
| 25 |
# Realiza a inferência
|
| 26 |
interpreter.set_tensor(input_details[0]['index'], image)
|
|
@@ -30,7 +32,6 @@ def predict(image):
|
|
| 30 |
# O resultado pode precisar ser ajustado conforme a saída do modelo
|
| 31 |
return output.tolist() # Retorna o resultado da inferência
|
| 32 |
|
| 33 |
-
|
| 34 |
# Configurando a interface Gradio
|
| 35 |
interface = gr.Interface(fn=predict, inputs="image", outputs="label")
|
| 36 |
-
interface.launch()
|
|
|
|
| 11 |
input_details = interpreter.get_input_details()
|
| 12 |
output_details = interpreter.get_output_details()
|
| 13 |
|
|
|
|
| 14 |
def predict(image):
|
| 15 |
# Verifica se a imagem está no formato correto (RGB)
|
| 16 |
+
if isinstance(image, Image.Image): # Verifica se é uma imagem PIL
|
| 17 |
+
if image.mode != 'RGB':
|
| 18 |
+
image = image.convert('RGB')
|
| 19 |
+
|
| 20 |
+
# Redimensiona a imagem para o tamanho necessário pelo modelo
|
| 21 |
+
image = image.resize((224, 224)) # Ajuste para o tamanho que seu modelo exige
|
| 22 |
|
| 23 |
+
# Converte a imagem para um numpy array
|
| 24 |
+
image = np.array(image).astype(np.float32) # Converte para numpy array
|
| 25 |
+
image = np.expand_dims(image, axis=0) # Adiciona uma dimensão extra para representar o batch
|
|
|
|
| 26 |
|
| 27 |
# Realiza a inferência
|
| 28 |
interpreter.set_tensor(input_details[0]['index'], image)
|
|
|
|
| 32 |
# O resultado pode precisar ser ajustado conforme a saída do modelo
|
| 33 |
return output.tolist() # Retorna o resultado da inferência
|
| 34 |
|
|
|
|
| 35 |
# Configurando a interface Gradio
|
| 36 |
interface = gr.Interface(fn=predict, inputs="image", outputs="label")
|
| 37 |
+
interface.launch(share=True) # Adiciona 'share=True' para gerar o link público
|