GabrielMI commited on
Commit
5638254
·
verified ·
1 Parent(s): 582dd13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ repo_id = "GabrielMI/entregable3"
14
 
15
 
16
  learner = from_pretrained_fastai(repo_id)
17
- labels = ["C++", "Fortran", "Go", "Java", "Python"]
18
 
19
  def predict(text):
20
  pred, pred_idx, probs = learner.predict(text)
 
14
 
15
 
16
  learner = from_pretrained_fastai(repo_id)
17
+ labels = ["C", "C++", "Fortran", "Go", "Java", "JavaScript", "Python", "Ruby"]
18
 
19
  def predict(text):
20
  pred, pred_idx, probs = learner.predict(text)