Update app.py
Browse files
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
|
| 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()
|