GDDO commited on
Commit
e0855e4
·
verified ·
1 Parent(s): 31d7cdc
Files changed (1) hide show
  1. app.py +10 -9
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.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
- image = np.array(image).astype(np.float32) # Converte a imagem para numpy array
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