Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,3 @@
|
|
1 |
-
# app.py
|
2 |
-
|
3 |
import gradio as gr
|
4 |
import torch
|
5 |
import numpy as np
|
@@ -14,15 +12,15 @@ import os
|
|
14 |
import traceback
|
15 |
|
16 |
# Descomprimir el modelo si no se ha descomprimido a煤n
|
17 |
-
if not os.path.exists("saved_model"):
|
18 |
with zipfile.ZipFile("saved_model.zip", "r") as zip_ref:
|
19 |
-
zip_ref.extractall("
|
20 |
|
21 |
# Cargar modelo ISIC con TensorFlow desde el directorio correcto
|
22 |
from keras.layers import TFSMLayer
|
23 |
|
24 |
try:
|
25 |
-
model_isic = TFSMLayer("
|
26 |
except Exception as e:
|
27 |
print("\U0001F534 Error al cargar el modelo ISIC con TFSMLayer:", e)
|
28 |
raise
|
@@ -103,7 +101,7 @@ def analizar_lesion_combined(img):
|
|
103 |
<tr><td>馃К Fast.ai (clasificaci贸n)</td><td><b>{pred_fast_type}</b></td><td>N/A</td></tr>
|
104 |
<tr><td>鈿狅笍 Fast.ai (malignidad)</td><td><b>{"Maligno" if prob_malignant > 0.5 else "Benigno"}</b></td><td>{prob_malignant:.1%}</td></tr>
|
105 |
<tr><td>馃敩 ISIC TensorFlow</td><td><b>{pred_class_isic}</b></td><td>{confidence_isic:.1%}</td></tr>
|
106 |
-
</table><br><b
|
107 |
|
108 |
cancer_risk_score = sum(probs_vit[i] * RISK_LEVELS[i]['weight'] for i in range(7))
|
109 |
if prob_malignant > 0.7 or cancer_risk_score > 0.6:
|
@@ -138,3 +136,4 @@ demo = gr.Interface(
|
|
138 |
if __name__ == "__main__":
|
139 |
demo.launch()
|
140 |
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
import numpy as np
|
|
|
12 |
import traceback
|
13 |
|
14 |
# Descomprimir el modelo si no se ha descomprimido a煤n
|
15 |
+
if not os.path.exists("saved_model.pb"):
|
16 |
with zipfile.ZipFile("saved_model.zip", "r") as zip_ref:
|
17 |
+
zip_ref.extractall(".")
|
18 |
|
19 |
# Cargar modelo ISIC con TensorFlow desde el directorio correcto
|
20 |
from keras.layers import TFSMLayer
|
21 |
|
22 |
try:
|
23 |
+
model_isic = TFSMLayer(".", call_endpoint="serving_default")
|
24 |
except Exception as e:
|
25 |
print("\U0001F534 Error al cargar el modelo ISIC con TFSMLayer:", e)
|
26 |
raise
|
|
|
101 |
<tr><td>馃К Fast.ai (clasificaci贸n)</td><td><b>{pred_fast_type}</b></td><td>N/A</td></tr>
|
102 |
<tr><td>鈿狅笍 Fast.ai (malignidad)</td><td><b>{"Maligno" if prob_malignant > 0.5 else "Benigno"}</b></td><td>{prob_malignant:.1%}</td></tr>
|
103 |
<tr><td>馃敩 ISIC TensorFlow</td><td><b>{pred_class_isic}</b></td><td>{confidence_isic:.1%}</td></tr>
|
104 |
+
</table><br><b>馃 Recomendaci贸n autom谩tica:</b><br>"""
|
105 |
|
106 |
cancer_risk_score = sum(probs_vit[i] * RISK_LEVELS[i]['weight'] for i in range(7))
|
107 |
if prob_malignant > 0.7 or cancer_risk_score > 0.6:
|
|
|
136 |
if __name__ == "__main__":
|
137 |
demo.launch()
|
138 |
|
139 |
+
|