LoloSemper commited on
Commit
254879b
verified
1 Parent(s): a0387fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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("saved_model")
20
 
21
  # Cargar modelo ISIC con TensorFlow desde el directorio correcto
22
  from keras.layers import TFSMLayer
23
 
24
  try:
25
- model_isic = TFSMLayer("saved_model/saved_model", call_endpoint="serving_default")
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>馃Ш Recomendaci贸n autom谩tica:</b><br>"""
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
+