ivferns commited on
Commit
a333f6c
·
verified ·
1 Parent(s): d4c932a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
- from fastai.vision.all import load_learner
3
 
4
- # Carga del modelo exportado
5
- learn = load_learner('export.pkl')
6
  labels = learn.dls.vocab
7
 
8
  # Función de predicción
@@ -27,4 +27,4 @@ app = gr.Interface(
27
  )
28
 
29
  if __name__ == "__main__":
30
- app.launch(server_name="0.0.0.0", server_port=7860)
 
1
  import gradio as gr
2
+ from huggingface_hub import from_pretrained_fastai
3
 
4
+ # Carga del modelo desde HuggingFace Hub
5
+ learn = from_pretrained_fastai("ivferns/futurama")
6
  labels = learn.dls.vocab
7
 
8
  # Función de predicción
 
27
  )
28
 
29
  if __name__ == "__main__":
30
+ app.launch(server_name="0.0.0.0", server_port=7860)