almangad commited on
Commit
021c565
·
verified ·
1 Parent(s): 88697a2

arreglo 4

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,7 +6,7 @@ repo_id = "almangad/entregable3"
6
 
7
  learner = from_pretrained_fastai(repo_id)
8
 
9
- labels = list(learner.dls.vocab[1])
10
 
11
  def predict_text(texto):
12
  pred, pred_idx, probs = learner.predict(texto)
@@ -20,4 +20,7 @@ demo = gr.Interface(
20
  ["im feeling quite sad and sorry for myself but ill snap out of it soon"],
21
  ["i am feeling grouchy"]
22
  ],
23
- cache_examples=False
 
 
 
 
6
 
7
  learner = from_pretrained_fastai(repo_id)
8
 
9
+ labels = list(learner.dls.vocab)
10
 
11
  def predict_text(texto):
12
  pred, pred_idx, probs = learner.predict(texto)
 
20
  ["im feeling quite sad and sorry for myself but ill snap out of it soon"],
21
  ["i am feeling grouchy"]
22
  ],
23
+ cache_examples=False
24
+ )
25
+
26
+ demo.launch()