DHEIVER commited on
Commit
95bc528
·
1 Parent(s): c5932f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -14,6 +14,12 @@ with open("categories.json", "r") as json_file:
14
  categories = [entry["category"] for entry in categories_data]
15
  diagnoses = [entry["diagnosis"] for entry in categories_data]
16
 
 
 
 
 
 
 
17
  # Crie uma função para realizar a classificação
18
  def classify_image(image):
19
  try:
@@ -34,12 +40,14 @@ def classify_image(image):
34
  except Exception as e:
35
  return str(e)
36
 
37
- # Crie uma interface Gradio para carregar a imagem e exibir a classificação com diagnóstico
38
  iface = gr.Interface(
39
  fn=classify_image,
40
  inputs=gr.inputs.Image(), # Entrada de imagem
41
- outputs="text" # Saída de texto com a classe prevista e diagnóstico
 
 
42
  )
43
 
44
  # Inicie a interface Gradio
45
- iface.launch()
 
14
  categories = [entry["category"] for entry in categories_data]
15
  diagnoses = [entry["diagnosis"] for entry in categories_data]
16
 
17
+ # Descrição do modelo e seu objetivo em português
18
+ model_description = (
19
+ "Este modelo foi treinado para classificar imagens médicas do trato gastrointestinal humano em várias categorias "
20
+ "com diagnósticos associados. Ele pode ajudar a identificar condições médicas a partir de imagens."
21
+ )
22
+
23
  # Crie uma função para realizar a classificação
24
  def classify_image(image):
25
  try:
 
40
  except Exception as e:
41
  return str(e)
42
 
43
+ # Crie uma interface Gradio com descrição completa e título informativo em português
44
  iface = gr.Interface(
45
  fn=classify_image,
46
  inputs=gr.inputs.Image(), # Entrada de imagem
47
+ outputs="text", # Saída de texto com a classe prevista e diagnóstico
48
+ title="Sistema de Classificação de Anomalias Gastrointestinais por Imagem",
49
+ description=model_description
50
  )
51
 
52
  # Inicie a interface Gradio
53
+ iface.launch()