firuiz commited on
Commit
9cae9cb
·
1 Parent(s): 145ce34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -2,8 +2,10 @@ from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
  from fastai.text.all import *
4
 
5
- learner = load_learner("entrega")
6
- labels = list(range(0,150))
 
 
7
 
8
 
9
  def predict(text):
 
2
  import gradio as gr
3
  from fastai.text.all import *
4
 
5
+ repo_id = "firuiz/entregable3"
6
+
7
+ learner = load_learner(repo_id)
8
+ labels = learner.dls.vocab
9
 
10
 
11
  def predict(text):