Edupy commited on
Commit
3904ea9
·
1 Parent(s): 91b4f35

Subiendo modelo y app de Pokémon

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -1,15 +1,22 @@
1
 
2
  from fastai.vision.all import *
3
  import gradio as gr
 
4
 
5
- # 1. Definir funciones que el modelo necesita (obligatorio para load_learner)
6
- def get_x(i): return None # Solo necesitamos que existan
7
  def get_y(i): return None
8
 
 
 
 
 
 
 
9
  # 2. Configuración de entorno
10
  defaults.silent = True
11
 
12
- # 3. Cargar el modelo (ahora encontrará get_x y get_y)
13
  learn = load_learner('modelo_pokemon.pkl')
14
  categories = learn.dls.vocab
15
 
 
1
 
2
  from fastai.vision.all import *
3
  import gradio as gr
4
+ import sys
5
 
6
+ # 1. Definir funciones que el modelo necesita
7
+ def get_x(i): return None
8
  def get_y(i): return None
9
 
10
+ # Parche crítico: Forzamos que las funciones existan en el módulo principal
11
+ # Esto soluciona el error UnboundLocalError al cargar el pickle
12
+ import __main__
13
+ __main__.get_x = get_x
14
+ __main__.get_y = get_y
15
+
16
  # 2. Configuración de entorno
17
  defaults.silent = True
18
 
19
+ # 3. Cargar el modelo
20
  learn = load_learner('modelo_pokemon.pkl')
21
  categories = learn.dls.vocab
22