DHEIVER commited on
Commit
9762bfd
1 Parent(s): c12444a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -8
app.py CHANGED
@@ -35,22 +35,28 @@ def classify_image(inp):
35
  # Obtém a classe prevista
36
  predicted_class = class_names[np.argmax(prediction)]
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>".format),
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,
49
- allow_flagging=False,
50
  layout="vertical",
51
  capture_session=True,
52
- theme="default", # Define o tema para o padrão
53
- )
54
 
55
  # Inicie a interface Gradio
56
- iface.launch()
 
35
  # Obtém a classe prevista
36
  predicted_class = class_names[np.argmax(prediction)]
37
 
38
+ # Formate a saída em HTML
39
+ result_html = f"""
40
+ <h2>Resultado da Classificação</h2>
41
+ <p><strong>Categoria Predita:</strong> {predicted_class}</p>
42
+ """
43
 
44
+ return result_html
45
+
46
+ # Crie uma interface Gradio
47
  iface = gr.Interface(
48
  fn=classify_image,
49
  inputs=gr.inputs.Image(shape=(192, 256)),
50
+ outputs=gr.outputs.HTML(), # Saída formatada com HTML
51
  live=True,
52
  title="Classificador de Catarata",
53
  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.",
54
+ allow_screenshot=False,
55
+ allow_flagging=False,
56
  layout="vertical",
57
  capture_session=True,
58
+ theme="default",
59
+ )
60
 
61
  # Inicie a interface Gradio
62
+ iface.launch()