joortif commited on
Commit
b806a6e
·
verified ·
1 Parent(s): 4dba9b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -2,8 +2,6 @@ from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
  from fastai.vision.all import *
4
 
5
-
6
-
7
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
8
  repo_id = "joortif/P1_AP_Tobacco"
9
 
@@ -12,7 +10,7 @@ labels = learner.dls.vocab
12
 
13
  # Definimos una función que se encarga de llevar a cabo las predicciones
14
  def predict(img):
15
- img = img.resize((128, 128))
16
  pred,pred_idx,probs = learner.predict(img)
17
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
18
 
 
2
  import gradio as gr
3
  from fastai.vision.all import *
4
 
 
 
5
  # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
6
  repo_id = "joortif/P1_AP_Tobacco"
7
 
 
10
 
11
  # Definimos una función que se encarga de llevar a cabo las predicciones
12
  def predict(img):
13
+ #img = img.resize((128, 128))
14
  pred,pred_idx,probs = learner.predict(img)
15
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
16