DHEIVER commited on
Commit
eea64d8
1 Parent(s): 0584be9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -19,7 +19,7 @@ with custom_object_scope({'FixedDropout': FixedDropout}):
19
  loaded_model = tf.keras.models.load_model('modelo_treinado.h5')
20
 
21
  # Crie uma lista de classes
22
- class_names = ["Normal", "Cataract"]
23
 
24
  # Defina a função de classificação
25
  def classify_image(inp):
@@ -37,16 +37,17 @@ def classify_image(inp):
37
 
38
  return predicted_class
39
 
40
- # Formate a interface Gradio com HTML
41
  iface = gr.Interface(
42
  fn=classify_image,
43
  inputs=gr.inputs.Image(shape=(192, 256)),
44
  outputs=gr.outputs.HTML("<h2>Resultado da Classificação</h2><p><strong>Categoria Predita:</strong> {}</p>"), # Saída formatada com HTML
45
  live=True,
46
- title="Classificador de Imagens",
47
- description="Faça o upload de uma imagem e classifique sua categoria.",
48
  allow_screenshot=False, # Desativar a captura de tela
49
  allow_flagging=False, # Desativar a opção de sinalizar
 
50
  )
51
 
52
  # Inicie a interface Gradio
 
19
  loaded_model = tf.keras.models.load_model('modelo_treinado.h5')
20
 
21
  # Crie uma lista de classes
22
+ class_names = ["Normal", "Catarata"]
23
 
24
  # Defina a função de classificação
25
  def classify_image(inp):
 
37
 
38
  return predicted_class
39
 
40
+ # Formate a interface Gradio com HTML e informações em português
41
  iface = gr.Interface(
42
  fn=classify_image,
43
  inputs=gr.inputs.Image(shape=(192, 256)),
44
  outputs=gr.outputs.HTML("<h2>Resultado da Classificação</h2><p><strong>Categoria Predita:</strong> {}</p>"), # Saída formatada com HTML
45
  live=True,
46
+ title="Classificador de Catarata",
47
+ description="Este é um classificador de imagens médicas para detectar catarata. Faça o upload de uma imagem e clique em 'Classificar' para obter a categoria predita.",
48
  allow_screenshot=False, # Desativar a captura de tela
49
  allow_flagging=False, # Desativar a opção de sinalizar
50
+ gr.Interface.Label("Faça o upload de uma imagem para classificar:")
51
  )
52
 
53
  # Inicie a interface Gradio